ntaexams commited on
Commit
f9acf93
·
verified ·
1 Parent(s): cc8ac42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model = AutoModelForCausalLM.from_pretrained(
13
  model_file=MODEL_FILE,
14
  model_type="mistral",
15
  # gpu_layers=50 if torch.cuda.is_available() else 0, # Use GPU if available
16
- context_length=1024 # Reduce context length for faster response
17
  )
18
  print("Model loaded successfully.")
19
 
 
13
  model_file=MODEL_FILE,
14
  model_type="mistral",
15
  # gpu_layers=50 if torch.cuda.is_available() else 0, # Use GPU if available
16
+ context_length=256 # Reduce context length for faster response
17
  )
18
  print("Model loaded successfully.")
19