Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -49,9 +49,9 @@ caption="A cake on the table."
|
|
49 |
# conditioning scale
|
50 |
paintingnet_conditioning_scale=1.0
|
51 |
|
52 |
-
brushnet = BrushNetModel.from_pretrained(brushnet_path, torch_dtype=torch.
|
53 |
pipe = StableDiffusionBrushNetPipeline.from_pretrained(
|
54 |
-
base_model_path, brushnet=brushnet, torch_dtype=torch.
|
55 |
)
|
56 |
|
57 |
# speed up diffusion process with faster scheduler and memory optimization
|
|
|
49 |
# conditioning scale
|
50 |
paintingnet_conditioning_scale=1.0
|
51 |
|
52 |
+
brushnet = BrushNetModel.from_pretrained(brushnet_path, torch_dtype=torch.float32)
|
53 |
pipe = StableDiffusionBrushNetPipeline.from_pretrained(
|
54 |
+
base_model_path, brushnet=brushnet, torch_dtype=torch.float32, low_cpu_mem_usage=False
|
55 |
)
|
56 |
|
57 |
# speed up diffusion process with faster scheduler and memory optimization
|