|
|
|
@ -78,7 +78,7 @@ class Pipeline: |
|
|
|
use_tiny_vae=True, |
|
|
|
use_tiny_vae=True, |
|
|
|
device=device, |
|
|
|
device=device, |
|
|
|
dtype=torch_dtype, |
|
|
|
dtype=torch_dtype, |
|
|
|
t_index_list=[1,16], |
|
|
|
t_index_list=[2,16], |
|
|
|
# t_index_list=[1], |
|
|
|
# t_index_list=[1], |
|
|
|
frame_buffer_size=1, |
|
|
|
frame_buffer_size=1, |
|
|
|
width=params.width, |
|
|
|
width=params.width, |
|
|
|
@ -101,7 +101,7 @@ class Pipeline: |
|
|
|
self.stream.prepare( |
|
|
|
self.stream.prepare( |
|
|
|
prompt=default_prompt, |
|
|
|
prompt=default_prompt, |
|
|
|
negative_prompt=default_negative_prompt, |
|
|
|
negative_prompt=default_negative_prompt, |
|
|
|
num_inference_steps=30, |
|
|
|
num_inference_steps=50, |
|
|
|
guidance_scale=0.9, |
|
|
|
guidance_scale=0.9, |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
|
