jijivski commited on
Commit
b9fc40e
1 Parent(s): d60b1f5

if 'model_cache' not in args.__dict__: # args.model_cache=args.model args.model_cache=None

Browse files
Files changed (1) hide show
  1. get_loss/get_loss_hf.py +2 -1
get_loss/get_loss_hf.py CHANGED
@@ -257,7 +257,8 @@ def run_get_loss(args=None):
257
  args.model='microsoft/phi-2'
258
 
259
  if 'model_cache' not in args.__dict__:
260
- args.model_cache=args.model
 
261
 
262
  # args = parser.parse_args()
263
 
 
257
  args.model='microsoft/phi-2'
258
 
259
  if 'model_cache' not in args.__dict__:
260
+ # args.model_cache=args.model
261
+ args.model_cache=None
262
 
263
  # args = parser.parse_args()
264