RealVis_v5.0_BF16_E

Running on Zero

ford442 commited on Dec 18, 2024

Commit

b1c614f

1 Parent(s): a51b2ce

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ def load_and_prepare_model(model_id):
     pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0")
     pipe = StableDiffusionXLPipeline.from_pretrained(
         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),

     pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0")
     pipe = StableDiffusionXLPipeline.from_pretrained(
         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),