mirror of https://github.com/THUDM/ChatGLM-6B
添加接口文档的注释
parent
02d4d3dd2c
commit
e63afa6f45
2
api.py
2
api.py
|
@ -36,7 +36,7 @@ def torch_gc():
|
||||||
app = FastAPI(lifespan=lifespan)
|
app = FastAPI(lifespan=lifespan)
|
||||||
|
|
||||||
class Item(BaseModel):
|
class Item(BaseModel):
|
||||||
prompt: str
|
prompt: str = "你好"
|
||||||
history: list[tuple[str, str]] = [[]]
|
history: list[tuple[str, str]] = [[]]
|
||||||
max_length: int = 2048
|
max_length: int = 2048
|
||||||
top_p: float = 0.7
|
top_p: float = 0.7
|
||||||
|
|
Loading…
Reference in New Issue