Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ if torch.cuda.is_available():
|
|
37 |
use_safetensors=True,
|
38 |
torch_dtype=torch.float32
|
39 |
)
|
40 |
-
vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.
|
41 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
42 |
base_model_id,
|
43 |
controlnet=controlnet,
|
|
|
37 |
use_safetensors=True,
|
38 |
torch_dtype=torch.float32
|
39 |
)
|
40 |
+
vae = AutoencoderKL.from_pretrained(vae_model_id, torch_dtype=torch.float32)
|
41 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
42 |
base_model_id,
|
43 |
controlnet=controlnet,
|