fix hflm pretrained in eval
Browse files- scripts/eval.py +1 -1
scripts/eval.py
CHANGED
|
@@ -238,7 +238,7 @@ def load_custom_model(args) -> HFLM:
|
|
| 238 |
|
| 239 |
# Wrap in HFLM
|
| 240 |
model = HFLM(
|
| 241 |
-
pretrained=
|
| 242 |
device=args.device,
|
| 243 |
batch_size=args.batch_size,
|
| 244 |
max_batch_size=args.max_batch_size,
|
|
|
|
| 238 |
|
| 239 |
# Wrap in HFLM
|
| 240 |
model = HFLM(
|
| 241 |
+
pretrained=args.model_path,
|
| 242 |
device=args.device,
|
| 243 |
batch_size=args.batch_size,
|
| 244 |
max_batch_size=args.max_batch_size,
|