Skip to content

Commit

Permalink
fix github actions
Browse files Browse the repository at this point in the history
  • Loading branch information
bmullick-amd authored Dec 18, 2024
1 parent 4f74d72 commit 36a1604
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions vllm/model_executor/models/t5.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,16 +139,16 @@ def load_tf_weights_in_t5(model, config, tf_checkpoint_path):
continue
elif scope_names[0] == "logits":
pointer = getattr(pointer, "lm_head", None)
elif scope_names[0] == "wi" and
len(scope_names) > 1 and
elif scope_names[0] == "wi" and \
len(scope_names) > 1 and \
scope_names[1].isdigit():
pointer = getattr(pointer, f"wi_{scope_names[1]}")
continue
else:
try:
pointer = getattr(pointer, scope_names[0])
except AttributeError:
log_name = '/'.join(name)
log_name = '/'.join(name)
logger.info("Skipping %s", log_name)
continue
if len(scope_names) >= 2:
Expand Down

0 comments on commit 36a1604

Please sign in to comment.