ford442 commited on
Commit
6df3a2f
·
1 Parent(s): 739bc8e

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",use_safetensors=True)
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",use_safetensors=True)
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),