Roastem commited on
Commit
59d67c7
1 Parent(s): 0fe5e7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ def load_model():
27
  # Load the locally downloaded model here, specifying the device
28
  llm = CTransformers(
29
  model="TheBloke/zephyr-7B-beta-GGUF",
30
- model_type="llama",
31
- max_new_tokens=1096,
32
  temperature=0.2,
33
  repetition_penalty=1.13,
34
  device=device # Set the device explicitly during model initialization
 
27
  # Load the locally downloaded model here, specifying the device
28
  llm = CTransformers(
29
  model="TheBloke/zephyr-7B-beta-GGUF",
30
+ model_type="mistral",
31
+ max_new_tokens=4384,
32
  temperature=0.2,
33
  repetition_penalty=1.13,
34
  device=device # Set the device explicitly during model initialization