Spaces:
Running
Running
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,
|