Gems234 commited on
Commit
f2f50f0
·
verified ·
1 Parent(s): e9bde8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -37,10 +37,10 @@ warnings.filterwarnings("ignore")
37
  print("⚡ Chargement du modèle avec llama.cpp...")
38
  llm = Llama(
39
  model_path=MODEL_PATH,
40
- n_ctx=4096,
41
  n_gpu_layers=0,
42
  n_threads=8,
43
- n_batch=512,
44
  verbose=False,
45
  use_mlock=True
46
  )
@@ -111,7 +111,7 @@ def send_message_stream(user_message, displayed_history, current_chat_name):
111
  stream = llm.create_completion(
112
  prompt=formatted_prompt,
113
  stream=True,
114
- max_tokens=1024,
115
  temperature=0.7,
116
  top_p=0.9,
117
  repeat_penalty=1.1,
 
37
  print("⚡ Chargement du modèle avec llama.cpp...")
38
  llm = Llama(
39
  model_path=MODEL_PATH,
40
+ n_ctx=2028,
41
  n_gpu_layers=0,
42
  n_threads=8,
43
+ #n_batch=512,
44
  verbose=False,
45
  use_mlock=True
46
  )
 
111
  stream = llm.create_completion(
112
  prompt=formatted_prompt,
113
  stream=True,
114
+ max_tokens=2028,
115
  temperature=0.7,
116
  top_p=0.9,
117
  repeat_penalty=1.1,