openai/CLIP
1
0
Fork 0
mirror of https://github.com/openai/CLIP.git synced 2026-01-26 15:29:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed another unused f-string (#276)

Browse Source
This commit is contained in:
sarveshwar-s
2022-07-27 21:42:37 +02:00
committed by GitHub
parent c5478aac7b
commit d50d76daa6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
clip/model.py
View File

@@ -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,