JackAILab commited on
Commit
b8a1fe8
1 Parent(s): 31287b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="
25
  pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
26
  base_model_path,
27
  torch_dtype=torch.float16,
28
- use_safetensors=True,
29
  variant="fp16"
30
  ).to(device)
31
 
 
25
  pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
26
  base_model_path,
27
  torch_dtype=torch.float16,
28
+ safety_checker=None, # use_safetensors=True,
29
  variant="fp16"
30
  ).to(device)
31