Add requirements.txt

2023-03-29 23:24:51 +09:00 · 2023-03-29 23:24:51 +09:00 · f7a66d7523
parent 0f1c288530
commit f7a66d7523
2 changed files with 7 additions and 1 deletions
--- a/handler.py
+++ b/handler.py
@ -20,4 +20,4 @@ class EndpointHandler:

        response, new_history = self.model.chat(self.tokenizer, inputs, history)

-        return [{"generated_text": response}]
+        return [{"generated_text": response}]
--- a/requirements.txt
+++ b/requirements.txt
@ -0,0 +1,6 @@
+protobuf>=3.19.5,<3.20.1
+transformers==4.26.1
+icetk
+cpm_kernels
+torch>=1.10
+gradio