style:format code

format code
This commit is contained in:
aries_ckt
2023-06-27 22:20:21 +08:00
parent 0878f3c5d4
commit 682b1468d1
47 changed files with 438 additions and 290 deletions

View File

@@ -27,7 +27,6 @@ from pilot.server.chat_adapter import get_llm_chat_adapter
CFG = Config()
class ModelWorker:
def __init__(self, model_path, model_name, device, num_gpus=1):
if model_path.endswith("/"):
@@ -106,6 +105,7 @@ worker = ModelWorker(
app = FastAPI()
from pilot.openapi.knowledge.knowledge_controller import router
app.include_router(router)
origins = [
@@ -119,9 +119,10 @@ app.add_middleware(
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"]
allow_headers=["*"],
)
class PromptRequest(BaseModel):
prompt: str
temperature: float