Commit 58865938 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add new export code, ci skip

parent 4db5f451

zoo/ptagger/model.py

+1 −1
Original line number Diff line number Diff line
@@ -79,7 +79,7 @@ def load_model(model_name: str = "tagger_v_2_2_7"): 
            filename=model.model_version_map[alt_model_name]['ckpt_name'],

        ), map_location="cpu")

        state_dicts_head = torch.load(hf_client.hf_hub_download(

            repo_id=model.model_version_map[alt_model_name],

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

            repo_type='model',

            filename=f'{model_name}.pth',

        ), map_location="cpu")