Spaces:
Running
on
L40S
Running
on
L40S
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ face_helper_1, face_helper_2, face_clip_model, face_main_model, eva_transform_me
|
|
47 |
|
48 |
|
49 |
# 2. Load Pipeline.
|
50 |
-
pipe = ConsisIDPipeline.from_pretrained(model_path,
|
51 |
|
52 |
# If you're using with lora, add this code
|
53 |
if lora_path:
|
|
|
47 |
|
48 |
|
49 |
# 2. Load Pipeline.
|
50 |
+
pipe = ConsisIDPipeline.from_pretrained(model_path, torch_dtype=dtype)
|
51 |
|
52 |
# If you're using with lora, add this code
|
53 |
if lora_path:
|