"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "fe49d5cbe0a5cc3d9ff9c99eb8278cdb0c48ea97"
Commit 9ded9e6b authored by myhloli's avatar myhloli
Browse files

refactor: simplify UnimernetModel constructor by removing unused cfg_path parameter

parent 51393aa8
...@@ -19,7 +19,7 @@ class MathDataset(Dataset): ...@@ -19,7 +19,7 @@ class MathDataset(Dataset):
class UnimernetModel(object): class UnimernetModel(object):
def __init__(self, weight_dir, cfg_path, _device_="cpu"): def __init__(self, weight_dir, _device_="cpu"):
from .unimernet_hf import UnimernetModel from .unimernet_hf import UnimernetModel
if _device_.startswith("mps"): if _device_.startswith("mps"):
self.model = UnimernetModel.from_pretrained(weight_dir, attn_implementation="eager") self.model = UnimernetModel.from_pretrained(weight_dir, attn_implementation="eager")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment