Skip to content

Commit

Permalink
Removed another unused f-string (#276)
Browse files Browse the repository at this point in the history
  • Loading branch information
sarveshwar-s authored Jul 27, 2022
1 parent c5478aa commit d50d76d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion clip/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -419,7 +419,7 @@ def build_model(state_dict: dict):
vocab_size = state_dict["token_embedding.weight"].shape[0]
transformer_width = state_dict["ln_final.weight"].shape[0]
transformer_heads = transformer_width // 64
transformer_layers = len(set(k.split(".")[2] for k in state_dict if k.startswith(f"transformer.resblocks")))
transformer_layers = len(set(k.split(".")[2] for k in state_dict if k.startswith("transformer.resblocks")))

model = CLIP(
embed_dim,
Expand Down

0 comments on commit d50d76d

Please sign in to comment.