Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse fileschange back to bobber/bigasp2
app.py
CHANGED
@@ -14,7 +14,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
14 |
|
15 |
# from bobber/bigasp2 to John6666/biglove-ponyv20-sdxl
|
16 |
good_vae = AutoencoderKL.from_pretrained("bobber/bigasp2", subfolder="vae", torch_dtype=dtype).to(device)
|
17 |
-
pipeline = StableDiffusionXLPipeline.from_pretrained("
|
18 |
|
19 |
MAX_SEED = np.iinfo(np.int32).max
|
20 |
|
|
|
14 |
|
15 |
# from bobber/bigasp2 to John6666/biglove-ponyv20-sdxl
|
16 |
good_vae = AutoencoderKL.from_pretrained("bobber/bigasp2", subfolder="vae", torch_dtype=dtype).to(device)
|
17 |
+
pipeline = StableDiffusionXLPipeline.from_pretrained("bobber/bigasp2", torch_dtype=dtype, vae=good_vae).to(device)
|
18 |
|
19 |
MAX_SEED = np.iinfo(np.int32).max
|
20 |
|