Code for the GIMMICK Paper
4
stars
9
commits
Python
primary language
Feb 21, 2025
updated

Tested only on Debian-based machines
Note that most models are loaded in FP16 or BF16 and Flash Attention 2. Make sure your GPU supports these features_
mamba env create -f environment.ymlmamba activate gimmick && pip install flash-attn --no-build-isolationsrc/scripts/check_env.shIf the last step was successful, it should print something like this:
##################### PYTHON ENV 'gimmick' INFO START #####################
Python version: 3.12.8 | packaged by conda-forge | (main, Dec 5 2024, 14:24:40) [GCC 13.3.0]
PyTorch version: 2.5.1
CUDA available: True
CUDA version: 11.8
CUDA devices: 1
Flash Attention 2 Support: True
Transformers version: 4.48.1
##################### PYTHON ENV 'gimmick' INFO END #####################
Run PYTHONPATH=src python src/gimmick/main.py models to print the list of supported models.
9 commits
Python
98.6%
Shell
1.4%
Code for the GIMMICK Paper
4
stars
9
commits
Python
primary language
Feb 21, 2025
updated

Tested only on Debian-based machines
Note that most models are loaded in FP16 or BF16 and Flash Attention 2. Make sure your GPU supports these features_
mamba env create -f environment.ymlmamba activate gimmick && pip install flash-attn --no-build-isolationsrc/scripts/check_env.shIf the last step was successful, it should print something like this:
##################### PYTHON ENV 'gimmick' INFO START #####################
Python version: 3.12.8 | packaged by conda-forge | (main, Dec 5 2024, 14:24:40) [GCC 13.3.0]
PyTorch version: 2.5.1
CUDA available: True
CUDA version: 11.8
CUDA devices: 1
Flash Attention 2 Support: True
Transformers version: 4.48.1
##################### PYTHON ENV 'gimmick' INFO END #####################
Run PYTHONPATH=src python src/gimmick/main.py models to print the list of supported models.
9 commits
Python
98.6%
Shell
1.4%