diff --git a/img2img.py b/img2img.py index 229bcf6..a69edbf 100644 --- a/img2img.py +++ b/img2img.py @@ -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: diff --git a/start.bat b/start.bat index b64a6d2..5bd49d4 100644 --- a/start.bat +++ b/start.bat @@ -1,4 +1,4 @@ @echo off cd /d %~dp0 call .venv\Scripts\activate.bat -.venv\Scripts\python.exe app.py \ No newline at end of file +.venv\Scripts\python.exe main.py \ No newline at end of file