dstars commited on
Commit
1068f35
·
verified ·
1 Parent(s): 340942c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import torch
5
  import torch.backends.cudnn as cudnn
6
  import gradio as gr
7
- from lmdeploy.messages import TurbomindEngineConfig, ChatTemplateConfig
8
  from utils import load_json, init_logger
9
  from demo import ConversationalAgent, CustomTheme
10
 
@@ -34,14 +34,6 @@ def main():
34
  food_examples = load_json(FOOD_EXAMPLES)
35
 
36
  # Set the device to CPU explicitly
37
- device = 'cpu'
38
- backend_config = TurbomindEngineConfig(
39
- model_format='pytorch',
40
- device=device,
41
- tp=1, # tensor parallel degree
42
- session_len=8192, # 最大会话长度
43
- rope_scaling_factor=1.0
44
- )
45
  agent = ConversationalAgent(model_path=MODEL_PATH,
46
  outputs_dir=OUTPUT_PATH,
47
  device=device,
 
4
  import torch
5
  import torch.backends.cudnn as cudnn
6
  import gradio as gr
7
+
8
  from utils import load_json, init_logger
9
  from demo import ConversationalAgent, CustomTheme
10
 
 
34
  food_examples = load_json(FOOD_EXAMPLES)
35
 
36
  # Set the device to CPU explicitly
 
 
 
 
 
 
 
 
37
  agent = ConversationalAgent(model_path=MODEL_PATH,
38
  outputs_dir=OUTPUT_PATH,
39
  device=device,