Spaces:

ZhouZJ36DL
/

Multi-turn_Consistent_Image_Editing_FLUX.1-dev

Running on Zero

ZhouwqZJ commited on May 6

Commit

b635626

1 Parent(s): b53eff9

modified: src/flux/model.py

Files changed (1) hide show

src/flux/model.py CHANGED Viewed

@@ -3,7 +3,7 @@ from dataclasses import dataclass
 import torch
 from torch import Tensor, nn
-from flux.modules.layers import (DoubleStreamBlock, EmbedND, LastLayer,
                                  MLPEmbedder, SingleStreamBlock,
                                  timestep_embedding)

 import torch
 from torch import Tensor, nn
+from src.flux.modules.layers import (DoubleStreamBlock, EmbedND, LastLayer,
                                  MLPEmbedder, SingleStreamBlock,
                                  timestep_embedding)