notebook-train / requirements.txt
alexkueck's picture
Update requirements.txt
ea4fe7e
raw
history blame contribute delete
249 Bytes
mercury>=2.2.6
gradio==3.11
torch
accelerate
transformers==4.29.1
sentencepiece #fast Tokenizer - necessary, wenn torch_dtype=torch.float16 angeschaltet ist bei funktion load_tokenizer_and_model
datasets
evaluate
scikit-learn
huggingface_hub