fixes for txt2img
This commit is contained in:
parent
60aa8ab4c0
commit
76bed9582b
|
@ -65,7 +65,7 @@ class SourceTxt2ImgStage(BaseStage):
|
||||||
else:
|
else:
|
||||||
# slice existing latents
|
# slice existing latents
|
||||||
latents = get_tile_latents(latents, dims, size)
|
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_type = params.get_valid_pipeline("txt2img")
|
||||||
pipe = load_pipeline(
|
pipe = load_pipeline(
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
from logging import getLogger
|
from logging import getLogger
|
||||||
from typing import Callable, List, Optional
|
from typing import Callable, List, Optional, Tuple
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import torch
|
import torch
|
||||||
|
|
Loading…
Reference in New Issue