Commit
•
200283f
1
Parent(s):
b8945f3
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
| 43 |
if torch.cuda.is_available():
|
| 44 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 45 |
pipe = DiffusionPipeline.from_pretrained(
|
| 46 |
-
"dataautogpt3/
|
| 47 |
vae=vae,
|
| 48 |
torch_dtype=torch.float16,
|
| 49 |
use_safetensors=True,
|
|
|
|
| 43 |
if torch.cuda.is_available():
|
| 44 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 45 |
pipe = DiffusionPipeline.from_pretrained(
|
| 46 |
+
"dataautogpt3/ProteusV0.2",
|
| 47 |
vae=vae,
|
| 48 |
torch_dtype=torch.float16,
|
| 49 |
use_safetensors=True,
|