zehaoj commited on
Commit
da55ae1
·
verified ·
1 Parent(s): 5938390

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -4,10 +4,10 @@ from pydantic import BaseModel
4
 
5
  #Model loading
6
  llm = AutoModelForCausalLM.from_pretrained("TheBloke/Toppy-M-7B-GGUF",
7
- model_file="toppy-m-7b.Q5_K_S.gguf",
8
  model_type="mistral",
9
  context_length=4096,
10
- temperature=0.3,
11
  gpu_layers=50)
12
 
13
 
 
4
 
5
  #Model loading
6
  llm = AutoModelForCausalLM.from_pretrained("TheBloke/Toppy-M-7B-GGUF",
7
+ model_file="toppy-m-7b.Q5_K_M.gguf",
8
  model_type="mistral",
9
  context_length=4096,
10
+ temperature=0.4,
11
  gpu_layers=50)
12
 
13