Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -225,10 +225,8 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
225 |
MAX_IMAGE_SIZE = 4096
|
226 |
|
227 |
if torch.cuda.is_available():
|
228 |
-
vae = StableDiffusionXLPipeline.from_single_file("https://huggingface.co/madebyollin/sdxl-vae-fp16-fix/blob/main/sdxl.vae.safetensors", torch_dtype=torch.float16)
|
229 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
230 |
"https://huggingface.co/Laxhar/noobai-XL-Vpred-0.6/blob/main/noobai-xl-vpred-v0.6.safetensors",
|
231 |
-
vae=vae,
|
232 |
use_safetensors=True,
|
233 |
torch_dtype=torch.float16,
|
234 |
add_watermarker=False
|
|
|
225 |
MAX_IMAGE_SIZE = 4096
|
226 |
|
227 |
if torch.cuda.is_available():
|
|
|
228 |
pipe = StableDiffusionXLPipeline.from_single_file(
|
229 |
"https://huggingface.co/Laxhar/noobai-XL-Vpred-0.6/blob/main/noobai-xl-vpred-v0.6.safetensors",
|
|
|
230 |
use_safetensors=True,
|
231 |
torch_dtype=torch.float16,
|
232 |
add_watermarker=False
|