AdrianM0 commited on
Commit
389ab02
·
verified ·
1 Parent(s): 269bf54

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -570,7 +570,7 @@ def load_model_and_tokenizers():
570
  tgt_vocab_size=config["tgt_vocab_size"],
571
  hparams_dict=config, # Pass the loaded config as hparams
572
  map_location=device, # Map model to the chosen device (CPU or CUDA)
573
- # strict=False, # Be less strict about matching keys, useful for PTL versions or minor changes
574
  )
575
 
576
  # Ensure model is on the correct device, in eval mode, and frozen
 
570
  tgt_vocab_size=config["tgt_vocab_size"],
571
  hparams_dict=config, # Pass the loaded config as hparams
572
  map_location=device, # Map model to the chosen device (CPU or CUDA)
573
+ strict=True, # Be less strict about matching keys, useful for PTL versions or minor changes
574
  )
575
 
576
  # Ensure model is on the correct device, in eval mode, and frozen