jianuo commited on
Commit
d08ef80
·
1 Parent(s): 91786f5

upload model

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +3 -2
app.py CHANGED
@@ -10,7 +10,7 @@ args = Namespace()
10
  arg_dict = {
11
  'run_gradio': True,
12
  'demo_public': False,
13
- 'model_name_or_path': 'Qwen/Qwen2-0.5B-Instruct',
14
 
15
  # 'model_name_or_path': 'Qwen/Qwen2-0.5B-Instruct-GGUF',
16
  'gguf_file': './qwen2-0_5b-instruct-q8_0.gguf', # 只有用gguf模型会用到,即model_name_or_path里含有gguf字符串才会用到
 
10
  arg_dict = {
11
  'run_gradio': True,
12
  'demo_public': False,
13
+ 'model_name_or_path': './model/Qwen2-0.5B-Instruct',
14
 
15
  # 'model_name_or_path': 'Qwen/Qwen2-0.5B-Instruct-GGUF',
16
  'gguf_file': './qwen2-0_5b-instruct-q8_0.gguf', # 只有用gguf模型会用到,即model_name_or_path里含有gguf字符串才会用到
requirements.txt CHANGED
@@ -4,9 +4,10 @@ torch
4
  transformers
5
  tokenizers
6
  accelerate
7
- text-generation==0.3.1
8
  optimum
9
  auto-gptq
10
  cpm_kernels
11
  bitsandbytes
12
- gradio==3.50.2
 
 
4
  transformers
5
  tokenizers
6
  accelerate
7
+ text-generation
8
  optimum
9
  auto-gptq
10
  cpm_kernels
11
  bitsandbytes
12
+ gradio==4.40.0
13
+ pydantic>=2.3