refactor file structure

rootonchair · Sep 2, 2024 · 00b5ffa · 00b5ffa
1 parent 0ec7dd5
commit 00b5ffa
Show file tree

Hide file tree

Showing 13 changed files with 15 additions and 18 deletions.
diff --git a/layer_diffuse/__init__.py b/layer_diffuse/__init__.py
diff --git a/loaders.py → layer_diffuse/loaders.py b/loaders.py → layer_diffuse/loaders.py
@@ -1,7 +1,6 @@
-import torch
 from safetensors.torch import load_file
 from diffusers.models.attention_processor import Attention, IPAdapterAttnProcessor, IPAdapterAttnProcessor2_0, AttnProcessor2_0
-from models import LoraLoader, AttentionSharingProcessor, IPAdapterAttnShareProcessor, AttentionSharingProcessor2_0, IPAdapterAttnShareProcessor2_0
+from .models import LoraLoader, AttentionSharingProcessor, IPAdapterAttnShareProcessor, AttentionSharingProcessor2_0, IPAdapterAttnShareProcessor2_0
 
 
 def merge_delta_weights_into_unet(pipe, delta_weights):

diff --git a/models/__init__.py → layer_diffuse/models/__init__.py b/models/__init__.py → layer_diffuse/models/__init__.py
diff --git a/models/attention_processors.py → layer_diffuse/models/attention_processors.py b/models/attention_processors.py → layer_diffuse/models/attention_processors.py
diff --git a/models/modules.py → layer_diffuse/models/modules.py b/models/modules.py → layer_diffuse/models/modules.py
diff --git a/utils.py → layer_diffuse/utils.py b/utils.py → layer_diffuse/utils.py
diff --git a/test_diffusers_bg_fg_cond.py b/test_diffusers_bg_fg_cond.py
@@ -6,9 +6,9 @@
 
 from diffusers import StableDiffusionPipeline
 
-from models import TransparentVAEDecoder
-from loaders import load_lora_to_unet
-from utils import rgba2rgbfp32, crop_and_resize_image
+from layer_diffuse.models import TransparentVAEDecoder
+from layer_diffuse.loaders import load_lora_to_unet
+from layer_diffuse.utils import rgba2rgbfp32, crop_and_resize_image
 
 
 

diff --git a/test_diffusers_fg_bg_cond.py b/test_diffusers_fg_bg_cond.py
@@ -7,9 +7,9 @@
 
 from diffusers import StableDiffusionPipeline
 
-from models import TransparentVAEDecoder
-from loaders import load_lora_to_unet
-from utils import rgba2rgbfp32, crop_and_resize_image
+from layer_diffuse.models import TransparentVAEDecoder
+from layer_diffuse.loaders import load_lora_to_unet
+from layer_diffuse.utils import rgba2rgbfp32, crop_and_resize_image
 
 
 

diff --git a/test_diffusers_fg_only.py b/test_diffusers_fg_only.py
@@ -4,8 +4,8 @@
 
 from diffusers import StableDiffusionPipeline
 
-from models import TransparentVAEDecoder
-from loaders import load_lora_to_unet
+from layer_diffuse.models import TransparentVAEDecoder
+from layer_diffuse.loaders import load_lora_to_unet
 
 
 

diff --git a/test_diffusers_fg_only_conv_sdxl.py b/test_diffusers_fg_only_conv_sdxl.py
@@ -1,12 +1,11 @@
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import torch
-from PIL import Image
 
 from diffusers import StableDiffusionXLPipeline
 
-from models import TransparentVAEDecoder
-from loaders import merge_delta_weights_into_unet
+from layer_diffuse.models import TransparentVAEDecoder
+from layer_diffuse.loaders import merge_delta_weights_into_unet
 
 
 if __name__ == "__main__":

diff --git a/test_diffusers_fg_only_sdxl.py b/test_diffusers_fg_only_sdxl.py
@@ -4,7 +4,7 @@
 
 from diffusers import StableDiffusionXLPipeline
 
-from models import TransparentVAEDecoder
+from layer_diffuse.models import TransparentVAEDecoder
 
 if __name__ == "__main__":
 

diff --git a/test_diffusers_fg_only_sdxl_img2img.py b/test_diffusers_fg_only_sdxl_img2img.py
@@ -5,7 +5,7 @@
 
 from diffusers import StableDiffusionXLInpaintPipeline
 
-from models import TransparentVAEDecoder, TransparentVAEEncoder
+from layer_diffuse.models import TransparentVAEDecoder, TransparentVAEEncoder
 
 
 if __name__ == "__main__":

diff --git a/test_diffusers_joint.py b/test_diffusers_joint.py
@@ -1,12 +1,11 @@
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import torch
-from PIL import Image
 
 from diffusers import StableDiffusionPipeline
 
-from models import TransparentVAEDecoder, LoraLoader, AttentionSharingProcessor
-from loaders import load_lora_to_unet
+from layer_diffuse.models import TransparentVAEDecoder
+from layer_diffuse.loaders import load_lora_to_unet