update param

main
reng 4 months ago
parent 3abb405413
commit 75df320b23
  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=[4, 16], t_index_list=[8, 32],
# 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=20, num_inference_steps=50,
guidance_scale=2, guidance_scale=0.2,
) )
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