cosmicman commited on
Commit
505226c
Β·
verified Β·
1 Parent(s): 0bb369d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -120,7 +120,7 @@ pipe.watermark = NoWatermark()
120
  refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
121
  base_model_path, # we found use base_model_path instead of refiner_model_path may get a better performance
122
  scheduler=scheduler,
123
- ,torch_dtype=torch.float16,
124
  use_safetensors=True
125
  ).to("cuda")
126
  refiner.watermark = NoWatermark()
 
120
  refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
121
  base_model_path, # we found use base_model_path instead of refiner_model_path may get a better performance
122
  scheduler=scheduler,
123
+ torch_dtype=torch.float16,
124
  use_safetensors=True
125
  ).to("cuda")
126
  refiner.watermark = NoWatermark()