diff --git a/img2img.py b/img2img.py index a69edbf..b783088 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=[4, 16], + t_index_list=[8, 32], # 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=20, - guidance_scale=2, + num_inference_steps=50, + guidance_scale=0.2, ) def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image: