Skip to content

Commit

Permalink
fix: lint with ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
drbh committed Dec 23, 2024
1 parent 575d973 commit 4c8f5cd
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -647,7 +647,7 @@ def __init__(self, prefix: str, config, weights):
)
self.model = FlashLlamaModel(prefix, config, weights)
if config.tie_word_embeddings:
suffix = f"model.embed_tokens"
suffix = "model.embed_tokens"
else:
suffix = "lm_head"

Expand Down

0 comments on commit 4c8f5cd

Please sign in to comment.