c2cite / requirements.txt
loadingy's picture
first push
51be264
torch>=2.3.0,<2.5.0
datasets
evaluate
accelerate
transformers>=4.44.0,<4.46.0
sentencepiece
huggingface_hub
scikit-learn
tiktoken
mixlora>=0.2.2,<0.3.0
gradio==4.38.1
peft==0.11.1
pandas
fire
black
black[jupyter]
isort
flake8
notebook
ipywidgets
nltk
rouge
rouge_score