update parameter

main
ULTRACOMBOS-DEV 2 months ago
parent b15a85a7a8
commit d481d0de4c
  1. 6
      img2img.py

@ -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=[2,16], t_index_list=[1, 8, 16, 24],
# t_index_list=[1], # t_index_list=[1],
frame_buffer_size=1, frame_buffer_size=1,
width=params.width, width=params.width,
@ -101,8 +101,8 @@ 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=50, num_inference_steps=32,
guidance_scale=0.9, guidance_scale=1.5,
) )
def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image: def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image:

Loading…
Cancel
Save