|
|
|
|
@ -78,8 +78,8 @@ class Pipeline: |
|
|
|
|
use_tiny_vae=True, |
|
|
|
|
device=device, |
|
|
|
|
dtype=torch_dtype, |
|
|
|
|
# t_index_list=[35, 45], |
|
|
|
|
t_index_list=[1], |
|
|
|
|
t_index_list=[4, 16], |
|
|
|
|
# t_index_list=[1], |
|
|
|
|
frame_buffer_size=1, |
|
|
|
|
width=params.width, |
|
|
|
|
height=params.height, |
|
|
|
|
@ -101,8 +101,8 @@ class Pipeline: |
|
|
|
|
self.stream.prepare( |
|
|
|
|
prompt=default_prompt, |
|
|
|
|
negative_prompt=default_negative_prompt, |
|
|
|
|
num_inference_steps=50, |
|
|
|
|
guidance_scale=1.0, |
|
|
|
|
num_inference_steps=20, |
|
|
|
|
guidance_scale=2, |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image: |
|
|
|
|
|