Updated requirements

README.md

@@ -12,10 +12,14 @@ Pueden clonar el repositorio con:
 
     $ git clone https://huggingface.co/crscardellino/flisol-cba-martin-fierro
 
-Y luego crean el entorno e instalan los requerimientos (el código está probado
-con Python 3.10, pero debería funcionar con Python >= 3.8):
+Y luego crean el entorno e instalan los requerimientos.
 
     $ python -m venv flisol-venv
     $ source ./flisol-venv/bin/activate
     (flisol-venv) $ pip install -r requirements.txt
 
+El código está probado con Python 3.10, pero debería funcionar con Python >=
+3.8. En los requerimientos está organizado para instalar
+[PyTorch](https://pytorch.org/) v2.0.0 para cpu, pero pueden ajustarlo para
+utilizar GPUs suponiendo que cumplan los requerimientos de CUDA.
+

requirements.txt

@@ -1,4 +1,5 @@
 --find-links https://download.pytorch.org/whl/torch_stable.html
+accelerate==0.18.0
 datasets==2.11.0
 jupyter==1.0.0
 numpy==1.24.2