ntaexams commited on
Commit
97e52cd
·
verified ·
1 Parent(s): 01ed2fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model = AutoModelForCausalLM.from_pretrained(
12
  MODEL_REPO,
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.")
 
12
  MODEL_REPO,
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.")