Skip to content
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

Upgrade torch to 2.5 #1080

Draft
wants to merge 28 commits into
base: mainline
Choose a base branch
from
Draft
Changes from 1 commit
Commits
Show all changes
28 commits
Select commit Hold shift + click to select a range
904545a
Upgrade torch to 2.5
papa99do Dec 23, 2024
5d25df8
a new base image
papa99do Dec 23, 2024
c59abae
change deps
papa99do Dec 23, 2024
200c9be
use the new base image
papa99do Dec 24, 2024
8ba27e7
ignore test_modalities_download
papa99do Dec 29, 2024
3473954
ignore test_model_cache_management
papa99do Dec 30, 2024
653e1b4
test encoding with onnx upgrade
papa99do Dec 30, 2024
86a6ef4
fix onnx version
papa99do Dec 30, 2024
6a0f8e0
fix onnx version again
papa99do Dec 30, 2024
91ce789
fix onnx version again
papa99do Dec 30, 2024
3b26d3c
remove onnxruntime-gpu
papa99do Dec 30, 2024
4585072
fix the sbert conversion issue
papa99do Dec 31, 2024
d1138f5
get back onnxruntime-gpu
papa99do Dec 31, 2024
136d01d
convert sbert to onnx op set 14
papa99do Dec 31, 2024
9cb47fc
test angular distance
papa99do Dec 31, 2024
0ef94c2
fix the test
papa99do Dec 31, 2024
9f46692
print out the angular distance info
papa99do Dec 31, 2024
21cdf0e
run all tests
papa99do Dec 31, 2024
6478f56
do not output
papa99do Dec 31, 2024
a941c76
test large clip model encoding
papa99do Jan 1, 2025
2b640ae
print out both angular distance and all close comparison result
papa99do Jan 1, 2025
01da5c5
trigger large model tests
papa99do Jan 2, 2025
8b42729
upgrade open_clip_torch to the latest version
papa99do Jan 2, 2025
db21e0b
Merge branch 'mainline' into yihan/torch-upgrade
papa99do Jan 3, 2025
3ee2f40
update base image, and run all tests
papa99do Jan 3, 2025
0d81e7c
rerun a flaky test
papa99do Jan 5, 2025
b679bd4
Merge branch 'mainline' into yihan/torch-upgrade
papa99do Jan 5, 2025
8e5828c
bake new iamge without ffmpeg-cuda
papa99do Jan 10, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
change deps
papa99do committed Dec 23, 2024
commit c59abaea1610ea9bfb62ff5512c1f4507f5b4cc2
1 change: 1 addition & 0 deletions .github/workflows/largemodel_unit_test_CI.yml
Original file line number Diff line number Diff line change
@@ -80,6 +80,7 @@ jobs:
uses: actions/checkout@v3
with:
repository: marqo-ai/marqo-base
ref: yihan/torch-upgrade
path: marqo-base

- name: Install dependencies
1 change: 1 addition & 0 deletions .github/workflows/unit_test_200gb_CI.yml
Original file line number Diff line number Diff line change
@@ -76,6 +76,7 @@ jobs:
uses: actions/checkout@v3
with:
repository: marqo-ai/marqo-base
ref: yihan/torch-upgrade
path: marqo-base

- name: Install dependencies