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