search

weaviate / multi2vec-clip-inference

Weaviate module inference code for the multi2vec-clip module
BSD 3-Clause "New" or "Revised" License
6 stars 7 forks source link

New pytorch 2.1+ to support Nvidia H100 gpu driver sm_90 #36

Open djjeffr opened 2 months ago

djjeffr commented 2 months ago

The current pytorch 2.0.1 doesn't support H100 sm_90 driver, can you update torch to at least 2.1.

djjeffr commented 1 month ago

It looks like the https://github.com/weaviate/t2v-transformers-models has been update, any idea when https://github.com/weaviate/multi2vec-clip-inference will be updated to a newer pytorch?