Spaces:
Runtime error
Runtime error
zhiweili
commited on
Commit
·
5c1adc5
1
Parent(s):
11b5282
fix cond image
Browse files
app_haircolor_inpaint_adapter_15.py
CHANGED
@@ -65,7 +65,7 @@ basepipeline = DiffusionPipeline.from_pretrained(
|
|
65 |
custom_pipeline="./pipelines/pipeline_sd_adapter_inpaint.py",
|
66 |
)
|
67 |
|
68 |
-
basepipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(basepipeline.scheduler.config)
|
69 |
|
70 |
basepipeline = basepipeline.to(DEVICE)
|
71 |
|
@@ -80,8 +80,8 @@ def image_to_image(
|
|
80 |
num_steps: int,
|
81 |
guidance_scale: float,
|
82 |
generate_size: int,
|
83 |
-
cond_scale1: float =
|
84 |
-
cond_scale2: float =
|
85 |
):
|
86 |
run_task_time = 0
|
87 |
time_cost_str = ''
|
|
|
65 |
custom_pipeline="./pipelines/pipeline_sd_adapter_inpaint.py",
|
66 |
)
|
67 |
|
68 |
+
# basepipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(basepipeline.scheduler.config)
|
69 |
|
70 |
basepipeline = basepipeline.to(DEVICE)
|
71 |
|
|
|
80 |
num_steps: int,
|
81 |
guidance_scale: float,
|
82 |
generate_size: int,
|
83 |
+
cond_scale1: float = 0.8,
|
84 |
+
cond_scale2: float = 0.8,
|
85 |
):
|
86 |
run_task_time = 0
|
87 |
time_cost_str = ''
|