fixes for txt2img
This commit is contained in:
parent
60aa8ab4c0
commit
76bed9582b
|
@ -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(
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue