Support for cpu/gpu version

2024-07-12 11:07:53 +00:00
parent eb0f28b3f6
commit a8901d2431
7 changed files with 760 additions and 631 deletions
@@ -1,18 +0,0 @@
-name: tfm_env
-
-channels:
-  - conda-forge
-  - defaults
-  - anaconda
-dependencies:
-  - jupyterlab
-  - pip
-  - python=3.10
-  - pip:
-    - huggingface_hub[cli]
-    - utilsforecast
-    - praxis
-    - paxml
-    - jax[cuda12]==0.4.26
-    - einshape
-    - scikit-learn