mirror of
https://github.com/snicolast/ComfyUI-IndexTTS2.git
synced 2026-05-01 04:01:36 +00:00
First Commit
This commit is contained in:
12
indextts/s2mel/hf_utils.py
Normal file
12
indextts/s2mel/hf_utils.py
Normal file
@@ -0,0 +1,12 @@
|
||||
import os
|
||||
from huggingface_hub import hf_hub_download
|
||||
|
||||
|
||||
def load_custom_model_from_hf(repo_id, model_filename="pytorch_model.bin", config_filename="config.yml"):
|
||||
os.makedirs("./checkpoints", exist_ok=True)
|
||||
model_path = hf_hub_download(repo_id=repo_id, filename=model_filename, cache_dir="./checkpoints")
|
||||
if config_filename is None:
|
||||
return model_path
|
||||
config_path = hf_hub_download(repo_id=repo_id, filename=config_filename, cache_dir="./checkpoints")
|
||||
|
||||
return model_path, config_path
|
||||
Reference in New Issue
Block a user