diff --git a/combined_diffusers_loader.py b/combined_diffusers_loader.py index f5bdf22..3a480ac 100644 --- a/combined_diffusers_loader.py +++ b/combined_diffusers_loader.py @@ -48,10 +48,11 @@ class CombinedDiffusersLoader: def load_models(self, sub_directory, clip_type="stable_diffusion"): base_path = self.get_base_path() print(f"CombinedDiffusersLoader: base_path: \n{base_path}") + print(f"CombinedDiffusersLoader: Sub_directory: \n{sub_directory}") sub_dir_path = os.path.join(base_path, sub_directory) - - model_type = self.detect_model_type(sub_dir_path) print(f"CombinedDiffusersLoader: Sub_dir_path: \n{sub_dir_path}") + model_type = self.detect_model_type(sub_dir_path) + vae_model = self.load_vae(sub_dir_path) unet_model = self.load_unet(sub_dir_path) clip_model = self.load_clip(sub_dir_path, clip_type, model_type)