ford442 commited on
Commit
995f31b
·
1 Parent(s): eaf43b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def load_and_prepare_model(model_id):
113
  pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0", torch_dtype=torch.float32)
114
  pipe = StableDiffusionXLPipeline.from_pretrained(
115
  model_id,
116
- # torch_dtype=torch.bfloat16,
117
  add_watermarker=False,
118
  # use_safetensors=True,
119
  # 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),
 
113
  pipeX = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V5.0", torch_dtype=torch.float32)
114
  pipe = StableDiffusionXLPipeline.from_pretrained(
115
  model_id,
116
+ torch_dtype=torch.bfloat16,
117
  add_watermarker=False,
118
  # use_safetensors=True,
119
  # 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),