Merge pull request THUDM#41 from secsilm/patch-1

fix skip_init import error
kwang1971 · Mar 15, 2023 · 999dd3c · 999dd3c
2 parents c55ecd8 + 3cbd413
commit 999dd3c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -2,3 +2,4 @@ protobuf>=3.19.5,<3.20.1
 transformers>=4.26.1
 icetk
 cpm_kernels
+torch>=1.10