AdrianM0 commited on
Commit
1c3db49
·
verified ·
1 Parent(s): 028e0b0

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +3 -7
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", PAD_IDX), # Use default if missing? Risky.
410
- "bos_token_id": config.get("bos_token_id", SOS_IDX), # Use default if missing? Risky.
411
- "eos_token_id": config.get("eos_token_id", EOS_IDX), # Use default if missing? Risky.
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: