DreamCanvas setup : to be reviewed
This commit is contained in:
13
On host/DreamCanvas/backend/models.py
Normal file
13
On host/DreamCanvas/backend/models.py
Normal file
@@ -0,0 +1,13 @@
|
||||
from pydantic import BaseModel
|
||||
|
||||
# Request model for LLM
|
||||
class LLMRequest(BaseModel):
|
||||
positive_prompt: str
|
||||
|
||||
# Request model for image generation
|
||||
class PromptRequest(BaseModel):
|
||||
positive_prompt: str
|
||||
negative_prompt: str
|
||||
steps: int = 25
|
||||
width: int = 512
|
||||
height: int = 512
|
||||
Reference in New Issue
Block a user