Commit
·
81115f7
1
Parent(s):
3ef3783
Refactor FluxImg2ImgReq class to make image and strength attributes optional
Browse files
app.py
CHANGED
|
@@ -117,15 +117,15 @@ class FluxReq(BaseModel):
|
|
| 117 |
|
| 118 |
|
| 119 |
class FluxImg2ImgReq(FluxReq):
|
| 120 |
-
image: Image.Image
|
| 121 |
-
strength: float = 1.0
|
| 122 |
|
| 123 |
class Config:
|
| 124 |
arbitrary_types_allowed=True
|
| 125 |
|
| 126 |
|
| 127 |
class FluxInpaintReq(FluxImg2ImgReq):
|
| 128 |
-
mask_image: Image.Image
|
| 129 |
|
| 130 |
class Config:
|
| 131 |
arbitrary_types_allowed=True
|
|
|
|
| 117 |
|
| 118 |
|
| 119 |
class FluxImg2ImgReq(FluxReq):
|
| 120 |
+
image: Optional[Image.Image]
|
| 121 |
+
strength: Optional[float] = 1.0
|
| 122 |
|
| 123 |
class Config:
|
| 124 |
arbitrary_types_allowed=True
|
| 125 |
|
| 126 |
|
| 127 |
class FluxInpaintReq(FluxImg2ImgReq):
|
| 128 |
+
mask_image: Optional[Image.Image]
|
| 129 |
|
| 130 |
class Config:
|
| 131 |
arbitrary_types_allowed=True
|