emotion_classifier / requirements.txt
fioriclass's picture
suppression cupy et cudf en doublon
fa9c546
raw
history blame contribute delete
204 Bytes
hydra-core
omegaconf
mlflow
pydantic
scikit-learn
scipy
numpy
pandas
optuna
ray[tune]
transformers
torch
datasets
seaborn
matplotlib
cupy-cuda12x
cuml-cu12==25.2.*
cudf-cu12==25.2.*
torchvision
accelerate