zhiweili commited on
Commit
c22253e
1 Parent(s): 5c43d2c

add variant

Browse files
Files changed (1) hide show
  1. app_masked.py +1 -0
app_masked.py CHANGED
@@ -30,6 +30,7 @@ basepipeline = DiffusionPipeline.from_pretrained(
30
  torch_dtype=torch.float16,
31
  # use_safetensors=True,
32
  custom_pipeline="./pipelines/masked_stable_diffusion_xl_img2img.py",
 
33
  )
34
 
35
  basepipeline = basepipeline.to(DEVICE)
 
30
  torch_dtype=torch.float16,
31
  # use_safetensors=True,
32
  custom_pipeline="./pipelines/masked_stable_diffusion_xl_img2img.py",
33
+ variant="fp16",
34
  )
35
 
36
  basepipeline = basepipeline.to(DEVICE)