search

HelloVision / HelloMeme

The official HelloMeme GitHub site
https://songkey.github.io/hellomeme/
MIT License
102 stars 6 forks source link

After run the app, all models will be downloaded #9

Open lior007 opened 8 hours ago

lior007 commented 8 hours ago

what models? What are the file names? What folder do they go to?

also Every time I run the APP.PY file 4 files are downloaded to the folder "C:\Users\LIOR0.cache\huggingface\hub\models--songkey--hello_group_facemodel\snapshots\a0e34229533a463364929f39ee7e7d72e3d2d188

this is the files: hello_3dmm.onnx hello_arkit_blendshape.onnx hello_face_det.onnx hello_face_landmark.onnx

Should they stay there? Or should they be copied to another location? Why do they go down every time?

songkey commented 2 hours ago
  1. You can find the corresponding file under hellomeme/tools and manually specify the cache directory for the model. For example, you can modify https://github.com/HelloVision/HelloMeme/blob/7dd30d97f0f9590d077e84a668df9901d23da713/hellomeme/tools/hello_3dmm.py#L40 to self.deep3d_pred_net = create_onnx_session(hf_hub_download('songkey/hello_group_facemodel', filename='hello_3dmm.onnx', cache_dir='path/to/you/defined'), gpu_id=gpu_id)
  2. Once a local copy of the model file (.onnx) is available, huggingface_hub will only verify the file’s integrity on subsequent launches, so it will not re-download the file.

https://huggingface.co/docs/huggingface_hub/guides/download