Tuchuanhuhuhu commited on
Commit
d9e7dbc
·
1 Parent(s): 7620bdc

拆分requirements

Browse files
modules/models.py CHANGED
@@ -10,9 +10,6 @@ import requests
10
  import urllib3
11
  import platform
12
 
13
- from dataclasses import dataclass, field
14
- from transformers import HfArgumentParser
15
-
16
  from tqdm import tqdm
17
  import colorama
18
  from duckduckgo_search import ddg
@@ -269,11 +266,6 @@ class ChatGLM_Client(BaseLLMModel):
269
  yield response
270
 
271
 
272
- @dataclass
273
- class ChatbotArguments:
274
- pass
275
-
276
-
277
  class LLaMA_Client(BaseLLMModel):
278
  def __init__(
279
  self,
 
10
  import urllib3
11
  import platform
12
 
 
 
 
13
  from tqdm import tqdm
14
  import colorama
15
  from duckduckgo_search import ddg
 
266
  yield response
267
 
268
 
 
 
 
 
 
269
  class LLaMA_Client(BaseLLMModel):
270
  def __init__(
271
  self,
requirements.txt CHANGED
@@ -13,11 +13,4 @@ markdown
13
  PyPDF2
14
  pdfplumber
15
  pandas
16
- transformers
17
- torch
18
- icetk
19
- protobuf==3.19.0
20
- git+https://github.com/OptimalScale/LMFlow.git#egg=lmflow
21
- cpm-kernels
22
- sentence_transformers
23
  commentjson
 
13
  PyPDF2
14
  pdfplumber
15
  pandas
 
 
 
 
 
 
 
16
  commentjson
requirements_advanced.txt ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ transformers
2
+ torch
3
+ icetk
4
+ protobuf==3.19.0
5
+ git+https://github.com/OptimalScale/LMFlow.git#egg=lmflow
6
+ cpm-kernels
7
+ sentence_transformers