Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,6 +38,7 @@ controlnet = ControlNetModel.from_pretrained(controlnet_path, use_safetensors=Fa
|
|
| 38 |
# load SDXL pipeline
|
| 39 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
| 40 |
base_model_path,
|
|
|
|
| 41 |
controlnet=controlnet,
|
| 42 |
torch_dtype=torch.float16,
|
| 43 |
add_watermarker=False,
|
|
|
|
| 38 |
# load SDXL pipeline
|
| 39 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
| 40 |
base_model_path,
|
| 41 |
+
variant='fp16',
|
| 42 |
controlnet=controlnet,
|
| 43 |
torch_dtype=torch.float16,
|
| 44 |
add_watermarker=False,
|