zehaoj commited on
Commit
3ca9fbc
·
verified ·
1 Parent(s): 12a7089

revert back to llm

Browse files
Files changed (1) hide show
  1. main.py +9 -9
main.py CHANGED
@@ -3,20 +3,20 @@ from fastapi import FastAPI, Form
3
  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_M.gguf",
8
- # model_type="mistral",
9
- # context_length=4096,
10
- # temperature=0.6,
11
- # gpu_layers=50)
12
-
13
- llm = AutoModelForCausalLM.from_pretrained("TheDrummer/Moistral-11B-v3-GGUF",
14
- model_file="Moistral-11B-v3-Q6_K.gguf",
15
  model_type="mistral",
16
  context_length=4096,
17
  temperature=0.6,
18
  gpu_layers=50)
19
 
 
 
 
 
 
 
 
20
 
21
  #Pydantic object
22
  class validation(BaseModel):
 
3
  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_M.gguf",
 
 
 
 
 
 
 
8
  model_type="mistral",
9
  context_length=4096,
10
  temperature=0.6,
11
  gpu_layers=50)
12
 
13
+ # llm = AutoModelForCausalLM.from_pretrained("TheDrummer/Moistral-11B-v3-GGUF",
14
+ # model_file="Moistral-11B-v3-Q6_K.gguf",
15
+ # model_type="mistral",
16
+ # context_length=4096,
17
+ # temperature=0.6,
18
+ # gpu_layers=50)
19
+
20
 
21
  #Pydantic object
22
  class validation(BaseModel):