Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
|