modified: src/flux/modules/layers.py

src/flux/modules/layers.py

@@ -5,7 +5,7 @@ import torch
 from einops import rearrange
 from torch import Tensor, nn
 
-from flux.math import attention, rope, adaptive_attention
+from src.flux.math import attention, rope, adaptive_attention
 
 import os
 

src/flux/util.py

@@ -8,10 +8,10 @@ from huggingface_hub import hf_hub_download
 from imwatermark import WatermarkEncoder
 from safetensors.torch import load_file as load_sft
 
-from flux.model import Flux, FluxParams
-from flux.modules.autoencoder import AutoEncoder, AutoEncoderParams
-from flux.modules.conditioner import HFEmbedder
-from flux.sampling import unpack
+from src.flux.model import Flux, FluxParams
+from src.flux.modules.autoencoder import AutoEncoder, AutoEncoderParams
+from src.flux.modules.conditioner import HFEmbedder
+from src.flux.sampling import unpack
 
 
 @dataclass