Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files从远端加载的模型生成的手段不符合逾期
app.py
CHANGED
@@ -225,7 +225,7 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
225 |
MAX_IMAGE_SIZE = 4096
|
226 |
|
227 |
if torch.cuda.is_available():
|
228 |
-
vae = AutoencoderKL.from_pretrained("
|
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,
|
|
|
225 |
MAX_IMAGE_SIZE = 4096
|
226 |
|
227 |
if torch.cuda.is_available():
|
228 |
+
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", 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,
|