apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/source_txt2img.py
View File

@ -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(

2
api/onnx_web/chain/upscale_outpaint.py
View File

@ -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