Commit
·
8517f95
1
Parent(s):
3ce21df
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ dtype = torch.float16
|
|
| 9 |
device = torch.device("cuda:0")
|
| 10 |
|
| 11 |
pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", torch_dtype=dtype)
|
|
|
|
| 12 |
pipeline.to(device)
|
| 13 |
|
| 14 |
|
|
|
|
| 9 |
device = torch.device("cuda:0")
|
| 10 |
|
| 11 |
pipeline = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", torch_dtype=dtype)
|
| 12 |
+
pipeline.vae.register_to_config("force_upcast"=False)
|
| 13 |
pipeline.to(device)
|
| 14 |
|
| 15 |
|