Jodaro commited on
Commit
0e9e41e
·
verified ·
1 Parent(s): e9ddae9

Use ctransformers qwen model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,9 +8,9 @@ print("Loading model...")
8
  llm = AutoModelForCausalLM.from_pretrained(
9
  MODEL_REPO,
10
  model_file=MODEL_FILE,
11
- model_type="llama",
12
  gpu_layers=0,
13
- context_length=4096,
14
  )
15
 
16
  def respond(message: str, history: list[list[str]]) -> str:
 
8
  llm = AutoModelForCausalLM.from_pretrained(
9
  MODEL_REPO,
10
  model_file=MODEL_FILE,
11
+ model_type="qwen",
12
  gpu_layers=0,
13
+ context_length=2048,
14
  )
15
 
16
  def respond(message: str, history: list[list[str]]) -> str: