Spaces:
Build error
Build error
Commit
•
553ce54
1
Parent(s):
41e603d
Update DAVINCIDIFFtextgardio.py
Browse files- DAVINCIDIFFtextgardio.py +1 -1
DAVINCIDIFFtextgardio.py
CHANGED
|
@@ -113,7 +113,7 @@ def generate_image(prompt, negative_prompt, cfg_scale, steps, width, height, see
|
|
| 113 |
torch_dtype=torch.float16,
|
| 114 |
use_safetensors=True,
|
| 115 |
vae=vae
|
| 116 |
-
).to('
|
| 117 |
|
| 118 |
generator = torch.manual_seed(seed)
|
| 119 |
|
|
|
|
| 113 |
torch_dtype=torch.float16,
|
| 114 |
use_safetensors=True,
|
| 115 |
vae=vae
|
| 116 |
+
).to('cpu')
|
| 117 |
|
| 118 |
generator = torch.manual_seed(seed)
|
| 119 |
|