Upload folder using huggingface_hub
Browse files
app.py
CHANGED
@@ -566,8 +566,6 @@ def load_model_and_tokenizers():
|
|
566 |
# Use the actual vocab sizes and hparams from the loaded config
|
567 |
model = SmilesIupacLitModule.load_from_checkpoint(
|
568 |
checkpoint_path,
|
569 |
-
# Pass necessary __init__ args that might not be in saved hparams automatically
|
570 |
-
# Ensure these keys exist in your loaded 'config' dict after validation/mapping
|
571 |
src_vocab_size=config["actual_src_vocab_size"],
|
572 |
tgt_vocab_size=config["actual_tgt_vocab_size"],
|
573 |
hparams_dict=config, # Pass the loaded config as hparams
|
|
|
566 |
# Use the actual vocab sizes and hparams from the loaded config
|
567 |
model = SmilesIupacLitModule.load_from_checkpoint(
|
568 |
checkpoint_path,
|
|
|
|
|
569 |
src_vocab_size=config["actual_src_vocab_size"],
|
570 |
tgt_vocab_size=config["actual_tgt_vocab_size"],
|
571 |
hparams_dict=config, # Pass the loaded config as hparams
|