Spaces:
Sleeping
Sleeping
Enable GPU laer and CORS setting
Browse files- app/main.py +1 -1
- app/utils/chat_rag.py +1 -1
app/main.py
CHANGED
@@ -20,7 +20,7 @@ app = FastAPI()
|
|
20 |
# Set all origins to wildcard for simplicity, but we should limit this in production
|
21 |
app.add_middleware(
|
22 |
CORSMiddleware,
|
23 |
-
allow_origins=["
|
24 |
allow_credentials=True,
|
25 |
allow_methods=["*"],
|
26 |
allow_headers=["*"],
|
|
|
20 |
# Set all origins to wildcard for simplicity, but we should limit this in production
|
21 |
app.add_middleware(
|
22 |
CORSMiddleware,
|
23 |
+
allow_origins=["https://educonnect-debd1.web.app"],
|
24 |
allow_credentials=True,
|
25 |
allow_methods=["*"],
|
26 |
allow_headers=["*"],
|
app/utils/chat_rag.py
CHANGED
@@ -92,7 +92,7 @@ class LlamaModelSingleton:
|
|
92 |
cls._instance.llm = LlamaCpp(
|
93 |
#streaming = True,
|
94 |
model_path=model_path,
|
95 |
-
|
96 |
n_batch=512,
|
97 |
temperature=0.1,
|
98 |
top_p=1,
|
|
|
92 |
cls._instance.llm = LlamaCpp(
|
93 |
#streaming = True,
|
94 |
model_path=model_path,
|
95 |
+
n_gpu_layers=-1,
|
96 |
n_batch=512,
|
97 |
temperature=0.1,
|
98 |
top_p=1,
|