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 = ''
|