Update app.py
Browse files
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
|