Menyu commited on
Commit
b8060d1
1 Parent(s): 4786b82

Update app.py

Browse files

从远端加载的模型生成的手段不符合逾期

Files changed (1) hide show
  1. app.py +1 -1
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("stabilityai/stable-diffusion-xl-base-1.0", 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,
 
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,