adnansami1992sami commited on
Commit
4b39e3f
1 Parent(s): b2f04c1

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -10,7 +10,7 @@ from ctransformers import AutoModelForCausalLM
10
  from pydantic import BaseModel
11
  from typing import List, Dict, Any, Generator
12
 
13
- llm = AutoModelForCausalLM.from_pretrained("TheBloke/falcon-40b-instruct-GGML", model_file="falcon40b-instruct.ggmlv3.q2_K.bin",
14
  model_type="falcon", threads=8)
15
  app = fastapi.FastAPI(title="🦅Falcon 40B GGML (ggmlv3.q2_K)🦅")
16
  app.add_middleware(
 
10
  from pydantic import BaseModel
11
  from typing import List, Dict, Any, Generator
12
 
13
+ llm = AutoModelForCausalLM.from_pretrained("TheBloke/WizardLM-Uncensored-Falcon-40B-GGML", model_file="wizardlm-uncensored-falcon-40b.ggccv1.q2_k.bin",
14
  model_type="falcon", threads=8)
15
  app = fastapi.FastAPI(title="🦅Falcon 40B GGML (ggmlv3.q2_K)🦅")
16
  app.add_middleware(