-
Notifications
You must be signed in to change notification settings - Fork 2.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: bump openclip to v2.5.0 #859
Conversation
Note: be careful with fp16 |
Codecov Report
@@ Coverage Diff @@
## bump-openclip-v25 #859 +/- ##
======================================================
- Coverage 80.28% 40.71% -39.57%
======================================================
Files 22 25 +3
Lines 1633 1820 +187
======================================================
- Hits 1311 741 -570
- Misses 322 1079 +757
Flags with carried forward coverage won't be shown. Click here to find out more.
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
@OrangeSodahub Please make sure the changes will not change the embedding result for each model. |
To transfer the code, we merge into a new subbranch first #860 . |
* feat: bump openclip to v2.5.0 (#859) * feat: bump openclip to v2.5.0 * fix: conflicts * fix: default fp32 on cpu and fp16 on gpu * feat: add two new models * fix: remove debug * fix: add roberta models (test) * fix: model name xlm * fix: (wip) * fix: remove roberta model * fix: model name * fix: add :: to model name * fix: add transformers * fix: remove is_init_value * fix: remove transformers * fix: not use flash-attn on cpu * fix: add assert description * fix: housekeeping * fix: cleanup * fix: cleanup * fix: allow to set precision * fix: gpu-test * fix: add roberta model test * fix: dtype * fix: dtype * fix: remove optional * fix: housekeeping * fix: housekeeping * fix: typo * fix: refactor * fix: refactor * fix: refactor * fix: type hint * fix: address comments * fix: visiontransformer * fix: d_model and n_head * fix: texttransformer * fix: change model to fix oom gh action * fix: class init and param name * fix: black * chore: bump open-clip version Co-authored-by: ZiniuYu <[email protected]>
Goals : align with openclip v2.5.0
Changes :
roberta-ViT-B-32
xlm-roberta-base-ViT-B-32
;LayernormFp32
(originalLayernorm
handles fp16).(Default set: fp32 on cpu and fp16 on gpu);
CLIP
toTextTransformer
,VisionTransformer
and add_build_text_tower
_build_vision_tower
;