add pix2pix

2023-01-28 21:13:21 +08:00
parent 205170e1e5
commit 05e82598de
14 changed files with 389 additions and 28 deletions
--- a/lama_cleaner/schema.py
+++ b/lama_cleaner/schema.py
@@ -88,3 +88,8 @@ class Config(BaseModel):
    paint_by_example_seed: int = 42
    paint_by_example_match_histograms: bool = False
    paint_by_example_example_image: Image = None
+
+    # InstructPix2Pix
+    p2p_steps: int = 50
+    p2p_image_guidance_scale: float = 1.5
+    p2p_guidance_scale: float = 7.5