ford442 commited on
Commit
1271363
Β·
verified Β·
1 Parent(s): a4ef9f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -142,8 +142,15 @@ def load_and_prepare_model(model_id):
142
  #scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset =1)
143
  )
144
  #pipe.vae = AsymmetricAutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2').to(torch.bfloat16) # ,use_safetensors=True FAILS
 
145
  #pipe.vae = AutoencoderKL.from_pretrained('ford442/Juggernaut-XI-v11-fp32',subfolder='vae') # ,use_safetensors=True FAILS
 
 
 
146
  #pipe.vae.to(torch.bfloat16)
 
 
 
147
  '''
148
  scaling_factor (`float`, *optional*, defaults to 0.18215):
149
  The component-wise standard deviation of the trained latent space computed using the first batch of the
 
142
  #scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start=0.00085, beta_end=0.012, steps_offset =1)
143
  )
144
  #pipe.vae = AsymmetricAutoencoderKL.from_pretrained('cross-attention/asymmetric-autoencoder-kl-x-2').to(torch.bfloat16) # ,use_safetensors=True FAILS
145
+
146
  #pipe.vae = AutoencoderKL.from_pretrained('ford442/Juggernaut-XI-v11-fp32',subfolder='vae') # ,use_safetensors=True FAILS
147
+
148
+ #pipe.vae = AutoencoderKL.from_pretrained('stabilityai/sdxl-vae-bf16',subfolder='vae')
149
+ #pipe.vae = AutoencoderKL.from_pretrained('stabilityai/sdxl-vae',subfolder='vae',force_upcast=False,scaling_factor= 0.182158767676)
150
  #pipe.vae.to(torch.bfloat16)
151
+
152
+
153
+
154
  '''
155
  scaling_factor (`float`, *optional*, defaults to 0.18215):
156
  The component-wise standard deviation of the trained latent space computed using the first batch of the