ford442 commited on
Commit
5105f45
·
verified ·
1 Parent(s): 5dc3d54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -31,8 +31,7 @@ import numpy as np
31
  from PIL import Image
32
  import torch
33
  from diffusers import AutoencoderKL, StableDiffusionXLPipeline
34
- #from diffusers import EulerAncestralDiscreteScheduler
35
- from diffusers import EulerDiscreteScheduler
36
  from typing import Tuple
37
  import paramiko
38
  import datetime
@@ -151,8 +150,8 @@ def load_and_prepare_model():
151
  add_watermarker=False,
152
  token=HF_TOKEN,
153
  )
154
- sched = EulerDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',token=HF_TOKEN)
155
- #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',token=HF_TOKEN)
156
  #pipe.vae = vaeRV #.to(torch.bfloat16)
157
  pipe.scheduler = sched
158
  #pipe.vae.do_resize=False
 
31
  from PIL import Image
32
  import torch
33
  from diffusers import AutoencoderKL, StableDiffusionXLPipeline
34
+ from diffusers import EulerAncestralDiscreteScheduler
 
35
  from typing import Tuple
36
  import paramiko
37
  import datetime
 
150
  add_watermarker=False,
151
  token=HF_TOKEN,
152
  )
153
+ #sched = EulerDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',token=HF_TOKEN)
154
+ sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',token=HF_TOKEN)
155
  #pipe.vae = vaeRV #.to(torch.bfloat16)
156
  pipe.scheduler = sched
157
  #pipe.vae.do_resize=False