Upload folder using huggingface_hub
Browse files
app.py
CHANGED
@@ -406,9 +406,9 @@ def load_model_and_tokenizers():
|
|
406 |
"num_decoder_layers": config.get("num_decoder_layers"),
|
407 |
"dropout": config.get("dropout"),
|
408 |
"max_len": config.get("max_len"),
|
409 |
-
"pad_token_id": config.get("pad_token_id"
|
410 |
-
"bos_token_id": config.get("bos_token_id"
|
411 |
-
"eos_token_id": config.get("eos_token_id"
|
412 |
}
|
413 |
# Update config with potentially remapped values
|
414 |
config.update(config_key_mapping)
|
@@ -417,10 +417,6 @@ def load_model_and_tokenizers():
|
|
417 |
if missing_keys:
|
418 |
# Try to load defaults for token IDs if absolutely necessary, but warn heavily
|
419 |
defaults_used = []
|
420 |
-
if "pad_token_id" in missing_keys and 'PAD_IDX' in globals(): config["pad_token_id"] = PAD_IDX; defaults_used.append("pad_token_id")
|
421 |
-
if "bos_token_id" in missing_keys and 'SOS_IDX' in globals(): config["bos_token_id"] = SOS_IDX; defaults_used.append("bos_token_id")
|
422 |
-
if "eos_token_id" in missing_keys and 'EOS_IDX' in globals(): config["eos_token_id"] = EOS_IDX; defaults_used.append("eos_token_id")
|
423 |
-
|
424 |
# Re-check missing keys after attempting defaults
|
425 |
missing_keys = [key for key in required_keys if config.get(key) is None]
|
426 |
if missing_keys:
|
|
|
406 |
"num_decoder_layers": config.get("num_decoder_layers"),
|
407 |
"dropout": config.get("dropout"),
|
408 |
"max_len": config.get("max_len"),
|
409 |
+
"pad_token_id": config.get("pad_token_id"), # Use default if missing? Risky.
|
410 |
+
"bos_token_id": config.get("bos_token_id"), # Use default if missing? Risky.
|
411 |
+
"eos_token_id": config.get("eos_token_id"), # Use default if missing? Risky.
|
412 |
}
|
413 |
# Update config with potentially remapped values
|
414 |
config.update(config_key_mapping)
|
|
|
417 |
if missing_keys:
|
418 |
# Try to load defaults for token IDs if absolutely necessary, but warn heavily
|
419 |
defaults_used = []
|
|
|
|
|
|
|
|
|
420 |
# Re-check missing keys after attempting defaults
|
421 |
missing_keys = [key for key in required_keys if config.get(key) is None]
|
422 |
if missing_keys:
|