Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,10 +72,10 @@ class LazyLoadPipeline:
|
|
| 72 |
@timer_func
|
| 73 |
def setup_pipeline(self):
|
| 74 |
print("Setting up the pipeline...")
|
| 75 |
-
controlnet = ControlNetModel
|
| 76 |
"models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16
|
| 77 |
)
|
| 78 |
-
safety_checker =
|
| 79 |
model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
|
| 80 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
|
| 81 |
model_path,
|
|
|
|
| 72 |
@timer_func
|
| 73 |
def setup_pipeline(self):
|
| 74 |
print("Setting up the pipeline...")
|
| 75 |
+
controlnet = ControlNetModel(
|
| 76 |
"models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16
|
| 77 |
)
|
| 78 |
+
safety_checker = None
|
| 79 |
model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
|
| 80 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
|
| 81 |
model_path,
|