Spaces:
Runtime error
Runtime error
Joseph Catrambone
commited on
Commit
·
a1a7f32
1
Parent(s):
3dbb2cf
Specify revision when loading the HF model.
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ from mediapipe_face_common import generate_annotation
|
|
14 |
|
15 |
# Download the SD 1.5 model from HF
|
16 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
17 |
-
model_path = hf_hub_download(repo_id="CrucibleAI/ControlNetMediaPipeFace", filename="models/controlnet_sd21_laion_face_v2_full.ckpt", repo_type="model")
|
18 |
-
config_path = hf_hub_download(repo_id="CrucibleAI/ControlNetMediaPipeFace", filename="models/cldm_v21.yaml", repo_type="model")
|
19 |
model = create_model(config_path).cpu()
|
20 |
model.load_state_dict(load_state_dict(model_path, location=device))
|
21 |
model = model.to(device)
|
|
|
14 |
|
15 |
# Download the SD 1.5 model from HF
|
16 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
17 |
+
model_path = hf_hub_download(repo_id="CrucibleAI/ControlNetMediaPipeFace", filename="models/controlnet_sd21_laion_face_v2_full.ckpt", repo_type="model", revision="568dc2c9980572262d48cff1ef2a7e4a03fadeb6")
|
18 |
+
config_path = hf_hub_download(repo_id="CrucibleAI/ControlNetMediaPipeFace", filename="models/cldm_v21.yaml", repo_type="model", revision="568dc2c9980572262d48cff1ef2a7e4a03fadeb6")
|
19 |
model = create_model(config_path).cpu()
|
20 |
model.load_state_dict(load_state_dict(model_path, location=device))
|
21 |
model = model.to(device)
|