Charlie81 commited on
Commit
2b77d15
·
1 Parent(s): 927465d

fix hflm pretrained in eval

Browse files
Files changed (1) hide show
  1. 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=hf_model,
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,