diff --git a/.gitignore b/.gitignore index 309dc80..a1ae2db 100644 --- a/.gitignore +++ b/.gitignore @@ -152,3 +152,6 @@ configs/local/default.yaml # Aim logging .aim + +outputs +.DS_Store \ No newline at end of file diff --git a/README.md b/README.md index cebc6aa..5ba443f 100644 --- a/README.md +++ b/README.md @@ -40,7 +40,7 @@ If you want to use NewsRecLib with GPU, please ensure CUDA or cudatoolkit versio ```bash git clone https://github.com/andreeaiana/newsreclib.git cd newsreclib - conda create --name newsreclib_env python=3.8 + conda create --name newsreclib_env python=3.9 conda activate newsreclib_env pip install -e . ``` diff --git a/setup.py b/setup.py index 5563116..ff3fdcf 100644 --- a/setup.py +++ b/setup.py @@ -11,8 +11,9 @@ url="https://github.com/andreeaiana/newsreclib", install_requires=[ "colorcet==3.0.1", + "cmake==3.18.4", "hydra-core==1.3.2", - "lightning==2.0.4", + "lightning==2.2.1", "MulticoreTSNE==0.1", "numpy==1.25.0", "omegaconf==2.3.0", @@ -28,10 +29,10 @@ "scikit-learn==1.2.2", "seaborn==0.12.2", "sphinx==5.0.2", - "tokenizers==0.13.2", + "tokenizers==0.15.2", "torchmetrics==1.3.1", "tqdm==4.65.0", - "transformers==4.36.0", + "transformers==4.38.2", "wandb==0.15.3", "hydra-colorlog==1.2.0", "hydra-optuna-sweeper==1.2.0",