diff --git a/img2img.py b/img2img.py index 3db71cc..2a03422 100644 --- a/img2img.py +++ b/img2img.py @@ -78,7 +78,7 @@ class Pipeline: use_tiny_vae=True, device=device, dtype=torch_dtype, - t_index_list=[2,16], + t_index_list=[1, 8, 16, 24], # t_index_list=[1], frame_buffer_size=1, width=params.width, @@ -101,8 +101,8 @@ class Pipeline: self.stream.prepare( prompt=default_prompt, negative_prompt=default_negative_prompt, - num_inference_steps=50, - guidance_scale=0.9, + num_inference_steps=32, + guidance_scale=1.5, ) def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image: