Spaces:

Lifeinhockey
/

Diffusion_Models

Runtime error

App Files Files Community

Lifeinhockey commited on Feb 14

Commit

e7563df

verified ·

1 Parent(s): b65dcf6

Update app.py

Browse files

Files changed (1) hide show

app.py +4 -4

app.py CHANGED Viewed

@@ -206,7 +206,7 @@ def infer(
                 # Инициализация ControlNet
                 controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
-                controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float16)
                 generator = torch.Generator(device).manual_seed(seed)
@@ -294,7 +294,7 @@ def infer(
                     # Инициализация ControlNet
                     controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
-                    controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float16)
                     generator = torch.Generator(device).manual_seed(seed)
@@ -394,7 +394,7 @@ def infer(
                     print('control_mode = ', control_mode)
                     controlnet_model_path = "lllyasviel/sd-controlnet-canny"
-                    controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float16, use_safetensors=True)
                     generator = torch.Generator(device).manual_seed(seed)
@@ -434,7 +434,7 @@ def infer(
                         depth_map = get_depth_map(control_image, depth_estimator).unsqueeze(0).half().to(device)
                         controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
-                        controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.float16, use_safetensors=True)
                         generator = torch.Generator(device).manual_seed(seed)

                 # Инициализация ControlNet
                 controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
+                controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)
                 generator = torch.Generator(device).manual_seed(seed)
                     # Инициализация ControlNet
                     controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
+                    controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype)
                     generator = torch.Generator(device).manual_seed(seed)
                     print('control_mode = ', control_mode)
                     controlnet_model_path = "lllyasviel/sd-controlnet-canny"
+                    controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype, use_safetensors=True)
                     generator = torch.Generator(device).manual_seed(seed)
                         depth_map = get_depth_map(control_image, depth_estimator).unsqueeze(0).half().to(device)
                         controlnet_model_path = "lllyasviel/control_v11f1p_sd15_depth"
+                        controlnet = ControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch_dtype, use_safetensors=True)
                         generator = torch.Generator(device).manual_seed(seed)