ECUiVADE commited on
Commit
a47a94d
1 Parent(s): b22108e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
29
 
30
  # Function to load the LLaMA model
31
  def load_model():
32
- llm = Llama(model_path=model_file, model_type="mistral",n_gpu_layers=-1,n_ctx = 2048)
33
  return llm
34
 
35
  # Function to generate a unique identifier for each chat session
 
29
 
30
  # Function to load the LLaMA model
31
  def load_model():
32
+ llm = Llama(model_path=model_file, model_type="mistral",n_gpu_layers=100,n_ctx = 2048)
33
  return llm
34
 
35
  # Function to generate a unique identifier for each chat session