Spaces:
Sleeping
Sleeping
thomas-yanxin
commited on
Commit
·
9783e10
1
Parent(s):
93a24b1
fix bug
Browse files- chatllm.py +1 -1
- requirements.txt +1 -1
chatllm.py
CHANGED
|
@@ -70,7 +70,7 @@ class ChatLLM(LLM):
|
|
| 70 |
prompt: str,
|
| 71 |
stop: Optional[List[str]] = None) -> str:
|
| 72 |
|
| 73 |
-
if self.
|
| 74 |
import requests
|
| 75 |
|
| 76 |
group_id = os.getenv('group_id')
|
|
|
|
| 70 |
prompt: str,
|
| 71 |
stop: Optional[List[str]] = None) -> str:
|
| 72 |
|
| 73 |
+
if self.model == 'Minimax':
|
| 74 |
import requests
|
| 75 |
|
| 76 |
group_id = os.getenv('group_id')
|
requirements.txt
CHANGED
|
@@ -14,4 +14,4 @@ faiss-cpu
|
|
| 14 |
gradio
|
| 15 |
nltk
|
| 16 |
accelerate
|
| 17 |
-
|
|
|
|
| 14 |
gradio
|
| 15 |
nltk
|
| 16 |
accelerate
|
| 17 |
+
detectron2@git+https://github.com/facebookresearch/detectron2.git@v0.6#egg=detectron2
|