abetlen / abetlen/llama-cpp-python

Which Python + CUDA version do you recommend for LLaVA on Windows 10 Pro 64-bit?

Đang mở
#1,418 1 bình luận 0 reaction 0 người được giao Xem trên GitHub
Ngôn ngữ chính
Python
Star
10.6k
Fork
1.4k
Chỉ số merge pull request
Chỉ số pull request đang chờ

Mô tả

Hey, I would appreciate recommendation on the specific versions of Python + CUDA environment on Windows that have been tested by you guys and proven to work well with LLaVAv1.6 and LLaVAv1.6 with CUDA GPU support in the latest releases. Can you provide me with some suggestions?

I have noticed that the LLaVA projector works fine for me when using "cuBLAS", but encounters issues when using "CUDA".
I tried to run it on a Python 3.11.4 64-bit + CUDA 12.1 and then with the latest CUDA 12.4.1 environments with llama-cpp-python installed with the adequate wheels, and without wheels through CMAKE_ARGS = "-DLLAMA_CUDA=on" , but couldn't get either LLaVAv1.5 or LLaVAv1.6 projectors to work correctly on release versions above 0.2.64.

There has been a change I can't identify between the release of 0.2.64 and 0.2.65, most likely related to CUDA/cuBLAS or chat_handler, that breaks how projectors work for me.

I fought with it the whole yesterday and eventually just gave up and reverted back to 0.2.64. The project I'm working on is currently running Bakllava-1-7b or LLaVAv1.5 without issues in a Python 3.10.11 64-bit + CUDA 11.7 environment using "cuBLAS on" with release version 0.2.64. I would like to upgrade to LLaVAv.1.6, at some point in the future.

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Đánh giá

Issue này chưa được đánh giá.

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.