Spaces:
Running
Running
matthoffner
commited on
Commit
ยท
0472301
1
Parent(s):
997f4b1
Update main.py
Browse files
main.py
CHANGED
@@ -21,7 +21,7 @@ config = {
|
|
21 |
}
|
22 |
llm = AutoModelForCausalLM.from_pretrained('TheBloke/MPT-7B-Storywriter-GGML',
|
23 |
model_file='mpt-7b-storywriter.ggmlv3.q4_0.bin',
|
24 |
-
model_type='mpt'
|
25 |
app = fastapi.FastAPI()
|
26 |
app.add_middleware(
|
27 |
CORSMiddleware,
|
|
|
21 |
}
|
22 |
llm = AutoModelForCausalLM.from_pretrained('TheBloke/MPT-7B-Storywriter-GGML',
|
23 |
model_file='mpt-7b-storywriter.ggmlv3.q4_0.bin',
|
24 |
+
model_type='mpt')
|
25 |
app = fastapi.FastAPI()
|
26 |
app.add_middleware(
|
27 |
CORSMiddleware,
|