423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #5307

Browse Source
This commit is contained in:
hiyouga
2024-08-30 02:45:40 +08:00
parent 3382317e32
commit 8b588c7224
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/api/app.py
View File

@@ -57,7 +57,8 @@ async def lifespan(app: "FastAPI"): # collects GPU memory
def create_app(chat_model: "ChatModel") -> "FastAPI":
app = FastAPI(lifespan=lifespan)
root_path = os.environ.get("FASTAPI_ROOT_PATH", "")
app = FastAPI(lifespan=lifespan, root_path=root_path)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],