Skip to content

Commit

Permalink
downg gymnasium + upg likely non blocking
Browse files Browse the repository at this point in the history
  • Loading branch information
MaximilienLC authored Nov 1, 2024
1 parent ef14f29 commit e4d0e70
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ dependencies = [
"jaxtyping==0.2.34", # Torch tensor annotations
"nptyping==2.5.0", # Numpy array annotations
"einops==0.8.0", # Makes shape transformations more readable
"jupyterlab==4.2.5", # Jupyter notebook
"jupyterlab==4.3.0", # Jupyter notebook
"jupyterlab-code-formatter==3.0.2", # Jupyter code formatter
"ipywidgets==8.1.5", # Jupyter widgets
# Replaces `pytorch_lightning` w/ `lightning.pytorch` to adress
Expand All @@ -51,23 +51,23 @@ dependencies = [
# OPTIONAL for cneuromax/fitting/deeplearning/
"torchaudio==2.5.1+cu124", # Tensor manipulation on audio data
"torchvision==0.20.1+cu124", # Tensor manipulation on vision data
"transformers==4.46.0", # Pre-trained models published on Hugging Face
"transformers==4.46.1", # Pre-trained models published on Hugging Face
"diffusers==0.31.0", # Diffusion models published on Hugging Face
"timm==1.0.11", # Image models
"datasets==3.0.2", # Datasets published on Hugging Face
"x-transformers==1.40.11", # Transformer utils
"datasets==3.1.0", # Datasets published on Hugging Face
"x-transformers==1.42.5", # Transformer utils
"mambapy==1.2.0", # Mamba
"denoising-diffusion-pytorch==2.1.1", # Diffusion models
"vector-quantize-pytorch==1.18.5", # Vector quantization utils
"accelerate==1.0.1", # PEFT crashing w/ 0.30.0
"accelerate==1.1.0", # PEFT crashing w/ 0.30.0
"peft==0.13.2", # Fine-tuning
# fMRI libraries
"nibabel==5.3.2", # Neuroimaging format accessor
"nilearn==0.10.4", # ML for neuroimaging
# MANDATORY for cneuromax/fitting/neuroevolution/
"mpi4py==4.0.1", # Inter-process communication
"torchrl==0.6.0", # For reinforcement/imitation Learning tasks
"gymnasium[mujoco]==1.0.0", # RL/IL environments
"gymnasium[mujoco]==0.29.1", # RL/IL environments
"ordered-set==4.1.0", # Sets w/ deterministic order
# MANDATORY for docs/
"sphinx==8.1.3", # Documentation generator
Expand Down

0 comments on commit e4d0e70

Please sign in to comment.