RealVis_v5.0_BF16_IP_B

Running on Zero

ford442 commited on Dec 12, 2024

Commit

cff36d6

1 Parent(s): 8242afe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ def load_and_prepare_model(model_id):
      #  torch_dtype=torch.bfloat16,
         add_watermarker=False,
         use_safetensors=True,
-        vae=AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",repo_type=model,safety_checker=None),
         scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1)
     ).to(device=device, dtype=torch.bfloat16)
    # sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1,)

      #  torch_dtype=torch.bfloat16,
         add_watermarker=False,
         use_safetensors=True,
+        vae=AutoencoderKL.from_pretrained("BeastHF/MyBack_SDXL_Juggernaut_XL_VAE/MyBack_SDXL_Juggernaut_XL_VAE_V10(version_X).safetensors",repo_type='model',safety_checker=None),
         scheduler=EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1)
     ).to(device=device, dtype=torch.bfloat16)
    # sched = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config, beta_schedule="scaled_linear", beta_start =0.00085,beta_end =0.012,steps_offset =1,)