AdrianM0 commited on
Commit
a3ae8ba
·
verified ·
1 Parent(s): 346f885

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -571,7 +571,6 @@ def load_model_and_tokenizers():
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
- # REMOVED invalid argument: device="cpu",
575
  )
576
 
577
  # Ensure model is on the correct device, in eval mode, and frozen
 
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