search

ml4bio / RNA-FM

RNA foundation model
https://ml4bio.github.io/RNA-FM/
MIT License
190 stars 18 forks source link

KeyError: "'args' key not found in model_data" #16

Closed kevin-liuguang closed 2 weeks ago

kevin-liuguang commented 2 weeks ago

When I run the command python launch/predict.py --config="pretrained/ss_prediction.yml" --data_path="./data/examples/example.fasta" --save_dir="./results" --save_frequency 1, I get the error: KeyError: "'args' key not found in model_data".

def _load_model_and_alphabet_core_v1(model_data, theme="protein"):
  # 检查 'args' 是否存在
  if "args" not in model_data:
    raise KeyError("'args' key not found in model_data")
  alphabet = fm.Alphabet.from_architecture(model_data["args"].arch, theme)