Commit
•
6f44fea
1
Parent(s):
139f1d0
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 39 |
|
| 40 |
|
| 41 |
if torch.cuda.is_available():
|
| 42 |
-
pipe =
|
| 43 |
"fluently/Fluently-XL-v2",
|
| 44 |
torch_dtype=torch.float16,
|
| 45 |
use_safetensors=True,
|
|
|
|
| 39 |
|
| 40 |
|
| 41 |
if torch.cuda.is_available():
|
| 42 |
+
pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 43 |
"fluently/Fluently-XL-v2",
|
| 44 |
torch_dtype=torch.float16,
|
| 45 |
use_safetensors=True,
|