Commit 69617993 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): update them, ci skip

parent 0531ed03

zoo/ptagger/model.py

+1 −1
Original line number Diff line number Diff line
@@ -80,7 +80,7 @@ def load_model(model_name: str = "tagger_v_2_2_7"): 
            repo_type='model',

            filename=model.model_version_map[alt_model_name]['ckpt_name'],

        ), map_location="cpu")

        model_repo_id = model.model_version_map[model_name]['repo_id']

        model_repo_id = model.model_version_map[alt_model_name]['repo_id']

        model_file = f'{model_name}.pth'

        state_dicts_head = torch.load(hf_client.hf_hub_download(

            repo_id=model.model_version_map[alt_model_name]['repo_id'],