Joseph Catrambone commited on
Commit
a1a7f32
1 Parent(s): 3dbb2cf

Specify revision when loading the HF model.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)