matthoffner commited on
Commit
463d78b
1 Parent(s): bf31376

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -11,7 +11,7 @@ from fastapi.middleware.cors import CORSMiddleware
11
  from sse_starlette.sse import EventSourceResponse
12
  from anyio import create_memory_object_stream
13
  from anyio.to_thread import run_sync
14
- from transformers import AutoModelForCausalLM
15
  from pydantic import BaseModel
16
 
17
 
 
11
  from sse_starlette.sse import EventSourceResponse
12
  from anyio import create_memory_object_stream
13
  from anyio.to_thread import run_sync
14
+ from ctransformers import AutoModelForCausalLM
15
  from pydantic import BaseModel
16
 
17