1
0
Fork 0

fixes for txt2img

This commit is contained in:
Sean Sube 2023-07-09 22:28:23 -05:00
parent 60aa8ab4c0
commit 76bed9582b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions

View File

@ -65,7 +65,7 @@ class SourceTxt2ImgStage(BaseStage):
else:
# slice existing latents
latents = get_tile_latents(latents, dims, size)
pipe_width, pipe_height, _tile_size = dims
pipe_width, pipe_height = size
pipe_type = params.get_valid_pipeline("txt2img")
pipe = load_pipeline(

View File

@ -1,5 +1,5 @@
from logging import getLogger
from typing import Callable, List, Optional
from typing import Callable, List, Optional, Tuple
import numpy as np
import torch