@ -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: