reqs

requirements/conda.requirements.txt

@@ -152,7 +152,7 @@ sklearn==0.0.post5
 smmap==5.0.0
 sniffio==1.3.0
 soupsieve==2.4.1
--e git+https://github.com/wassname/discovering_latent_knowledge.git@43074f79cbb58c735ff1befab81f28811831509b#egg=src
+-e git+https://github.com/wassname/discovering_latent_knowledge.git@7b532b1b300797fcdf9eb4fc6f1ed5fbde5470db#egg=src
 stack-data==0.6.2
 starlette==0.22.0
 starsessions==1.3.0

requirements/environment.max.yaml

@@ -149,24 +149,31 @@ dependencies:
   - zstd=1.5.2=hfc55251_7
   - pip:
     - accelerate==0.21.0
+    - aiodns==3.0.0
     - aiohttp==3.8.5
+    - aiohttp-retry==2.8.3
     - aiosignal==1.3.1
+    - aleph-alpha-client==2.17.0
     - annotated-types==0.5.0
     - anyio==3.7.1
     - arrow==1.2.3
     - async-timeout==4.0.3
     - attrs==23.1.0
     - backoff==2.2.1
+    - baukit==0.0.1
     - beautifulsoup4==4.12.2
     - bitsandbytes==0.39.1
     - black==23.7.0
     - blessed==1.20.0
+    - blinker==1.6.2
+    - cachetools==5.3.1
+    - cffi==1.15.1
     - click==8.1.6
     - concept-erasure==0.1.0
     - contourpy==1.1.0
     - croniter==1.4.1
     - cycler==0.11.0
-    - datasets==2.14.4
+    - datasets==2.14.5
     - dateutils==0.6.12
     - deepdiff==6.3.1
     - dill==0.3.7
@@ -178,11 +185,14 @@ dependencies:
     - frozenlist==1.4.0
     - fsspec==2023.6.0
     - h11==0.14.0
+    - httpcore==0.18.0
+    - httpx==0.25.0
     - huggingface-hub==0.16.4
     - inquirer==3.1.3
     - ipywidgets==8.1.0
     - itsdangerous==2.1.2
     - joblib==1.3.2
+    - json-tricks==3.17.3
     - jupyterlab-widgets==3.0.8
     - kaleido==0.2.1
     - kiwisolver==1.4.4
@@ -201,12 +211,16 @@ dependencies:
     - pathspec==0.11.2
     - peft==0.4.0
     - plotly==5.14.1
+    - polars==0.18.15
     - pyarrow==12.0.1
+    - pycares==4.3.0
+    - pycparser==2.21
     - pydantic==2.0.3
     - pydantic-core==2.3.0
     - pyjwt==2.8.0
     - pynvml==11.5.0
     - pyparsing==3.0.9
+    - python-dotenv==1.0.0
     - python-editor==1.0.4
     - python-multipart==0.0.6
     - pytorch-lightning==2.0.6
@@ -225,9 +239,11 @@ dependencies:
     - soupsieve==2.4.1
     - starlette==0.27.0
     - starsessions==1.3.0
+    - tabulate==0.9.0
     - tenacity==8.2.2
     - threadpoolctl==3.2.0
     - tokenizers==0.13.3
+    - torchinfo==1.8.0
     - torchmetrics==1.0.3
     - tqdm==4.66.1
     - transformers==4.31.0
@@ -236,6 +252,7 @@ dependencies:
     - websocket-client==1.6.1
     - websockets==11.0.3
     - widgetsnbextension==4.0.8
+    - xformers==0.0.21
     - xxhash==3.3.0
     - yarl==1.9.2
 prefix: /home/ubuntu/mambaforge/envs/dlk3