Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -109,7 +109,7 @@ class TextToImageTool(Tool):
|
|
| 109 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 110 |
print(f"Using device: {device}")
|
| 111 |
self.pipe = DiffusionPipeline.from_pretrained(
|
| 112 |
-
"
|
| 113 |
torch_dtype=dtype
|
| 114 |
).to(device)
|
| 115 |
|
|
|
|
| 109 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 110 |
print(f"Using device: {device}")
|
| 111 |
self.pipe = DiffusionPipeline.from_pretrained(
|
| 112 |
+
"aiyouthalliance/Free-Image-Generation-CC0",
|
| 113 |
torch_dtype=dtype
|
| 114 |
).to(device)
|
| 115 |
|