redotvideo / mamba-chat

Mamba-Chat: A chat LLM based on the state-space model architecture 🐍

Apache License 2.0

911 stars 69 forks source link

Added check to confirm bfloat support #20

Closed Rohith04MVK closed 3 months ago

Rohith04MVK commented 10 months ago

15 #11

Eupham commented 10 months ago

See notebook. It still requires performing the below once requirements are installed but this fix works. !export LC_ALL="en_US.UTF-8" !export LD_LIBRARY_PATH="/usr/lib64-nvidia" !export LIBRARY_PATH="/usr/local/cuda/lib64/stubs" !ldconfig /usr/lib64-nvidia