Spaces:
Runtime error
Runtime error
adnansami1992sami
commited on
Commit
•
4b39e3f
1
Parent(s):
b2f04c1
Update api.py
Browse files
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/
|
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(
|