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 = 1.2,
84
- cond_scale2: float = 1.2,
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 = ''