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,
|