lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #5307

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
Show Stats 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=["*"],