ford442 commited on
Commit
be88b26
·
verified ·
1 Parent(s): c216b49

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) pipe = StableDiffusionXLPipeline.from_pretrained(
 
86
  'ford442/RealVisXL_V5.0_BF16',
87
  #torch_dtype=torch.bfloat16,
88
  token=HF_TOKEN,
 
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,
89
  token=HF_TOKEN,