summaryrefslogtreecommitdiff
path: root/srm/requirements.txt
diff options
context:
space:
mode:
Diffstat (limited to 'srm/requirements.txt')
-rw-r--r--srm/requirements.txt11
1 files changed, 11 insertions, 0 deletions
diff --git a/srm/requirements.txt b/srm/requirements.txt
new file mode 100644
index 0000000..8c90d6f
--- /dev/null
+++ b/srm/requirements.txt
@@ -0,0 +1,11 @@
+torch
+adam-atan2
+einops
+tqdm
+coolname
+pydantic
+argdantic
+wandb
+omegaconf
+hydra-core
+huggingface_hub