Upload folder using huggingface_hub
Browse files
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
|