unified queueing

This commit is contained in:
Yannic Kilcher
2023-02-11 01:31:25 +01:00
parent 76f7af0dfd
commit 48212079f4
7 changed files with 144 additions and 82 deletions
@@ -9,6 +9,10 @@ from . import protocol
class WorkerConfig(pydantic.BaseModel):
model_name: str = "distilgpt2"
@property
def compat_hash(self) -> str:
return f"{self.model_name}"
class WorkRequest(pydantic.BaseModel):
conversation: protocol.Conversation = pydantic.Field(..., repr=False)