main
reng 4 months ago
parent 036f4d8d94
commit 3abb405413
  1. 8
      img2img.py
  2. 2
      start.bat

@ -78,8 +78,8 @@ class Pipeline:
use_tiny_vae=True, use_tiny_vae=True,
device=device, device=device,
dtype=torch_dtype, dtype=torch_dtype,
# t_index_list=[35, 45], t_index_list=[4, 16],
t_index_list=[1], # t_index_list=[1],
frame_buffer_size=1, frame_buffer_size=1,
width=params.width, width=params.width,
height=params.height, height=params.height,
@ -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=20,
guidance_scale=1.0, guidance_scale=2,
) )
def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image: def predict(self, image: Image.Image, params: "Pipeline.InputParams") -> Image.Image:

@ -1,4 +1,4 @@
@echo off @echo off
cd /d %~dp0 cd /d %~dp0
call .venv\Scripts\activate.bat call .venv\Scripts\activate.bat
.venv\Scripts\python.exe app.py .venv\Scripts\python.exe main.py
Loading…
Cancel
Save