ford442 commited on
Commit
ee7d909
·
verified ·
1 Parent(s): c8e57ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,7 +82,8 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
82
 
83
  def load_and_prepare_model():
84
  vaeXL = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False).to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
85
- sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
 
86
  pipe = StableDiffusionXLPipeline.from_pretrained(
87
  'ford442/RealVisXL_V5.0_BF16',
88
  #torch_dtype=torch.bfloat16,
 
82
 
83
  def load_and_prepare_model():
84
  vaeXL = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None, use_safetensors=False).to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
85
+ #sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset=1,use_karras_sigmas=True)
86
+ sched = EulerAncestralDiscreteScheduler.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='scheduler',beta_schedule="scaled_linear")
87
  pipe = StableDiffusionXLPipeline.from_pretrained(
88
  'ford442/RealVisXL_V5.0_BF16',
89
  #torch_dtype=torch.bfloat16,