Skip to content

Commit

Permalink
Merge pull request #59 from tatsu-lab/rtaori-patch-1
Browse files Browse the repository at this point in the history
log bug fix in recover_model_weights.py
  • Loading branch information
rtaori committed Jul 4, 2023
2 parents 19fb391 + f1b552e commit 55e0b68
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pretrained_models/recover_model_weights.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def load_raw_model(model_dir, device="cpu"):
config_path = os.path.join(model_dir, "config.json")
config = json.load(open(config_path, "r"))
transformers_version = config["transformers_version"]
if transformers_version <= min_transformers_version:
if transformers_version < min_transformers_version:
logging.warning(
f"Your base LLaMA checkpoint is converted with transformers=={transformers_version}, "
f"but transformers>={min_transformers_version} is expected. "
Expand Down

0 comments on commit 55e0b68

Please sign in to comment.