utensil
/
llm-playground
connected to https://github.com/utensil/llm-playground.git


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
            # Assuming image: winglian/axolotl-runpod:main-cu118-2.0.0

# TODO: maybe no -U or make it optional
pip install -q -U bitsandbytes
pip install -q -U git+https://github.com/huggingface/transformers.git 
pip install -q -U git+https://github.com/huggingface/peft.git
pip install -q -U git+https://github.com/huggingface/accelerate.git
pip install -q datasets
pip install -q einops scipy wandb

# Need to change this before uncommenting
# export WANDB_PROJECT=runpod

# Might need to fix bitsandbytes
# WORKSPACE=${WORKSPACE:-"/workspace"}
# cd $WORKSPACE
# git clone https://github.com/TimDettmers/bitsandbytes
# cd bitsandbytes/
# CUDA_VERSION=118 make cuda11x
# python setup.py install
# python -m bitsandbytes