Update app.py
Browse files
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
|
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
|