refactor(SDXL): better reusability

ovuruska · Apr 30, 2024 · 60c181f · 60c181f
1 parent 833de66
commit 60c181f
Showing 1 changed file with 3 additions and 16 deletions.
diff --git a/deepinfra/types/sdxl/request.py b/deepinfra/types/sdxl/request.py
@@ -2,22 +2,9 @@
 
 from dataclasses import dataclass
 
+from deepinfra.types.text_to_image import TextToImageRequest
+
 
 @dataclass
 class SdxlRequest:
-    prompt: str
-    negative_prompt: Optional[str] = None
-    image: Optional[str] = None
-    mask: Optional[str] = None
-    width: Optional[int] = None
-    height: Optional[int] = None
-    num_outputs: Optional[int] = None
-    scheduler: Optional[str] = None
-    num_inference_steps: Optional[int] = None
-    guidance_scale: Optional[float] = None
-    prompt_strength: Optional[float] = None
-    seed: Optional[int] = None
-    refine: Optional[str] = None
-    high_noise_frac: Optional[float] = None
-    refine_steps: Optional[int] = None
-    apply_watermark: Optional[bool] = True
+    input: TextToImageRequest