github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[SD] Reset args.img_path to None in txt2img to avoid vae_encode

Browse Source 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2023-02-21 18:08:53 +05:30
parent b63fb36dc0
commit 4233218629
3 changed files with 4 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
apps/stable_diffusion/scripts/img2img.py
View File

@@ -104,7 +104,7 @@ def img2img_inf(
width,
device,
)
if config_obj != new_config_obj:
if not img2img_obj or config_obj != new_config_obj:
config_obj = new_config_obj
args.precision = precision
args.batch_size = batch_size
@@ -139,9 +139,6 @@ def img2img_inf(
low_cpu_mem_usage=args.low_cpu_mem_usage,
)
if not img2img_obj:
sys.exit("text to image pipeline must not return a null value")
img2img_obj.scheduler = schedulers[scheduler]
start_time = time.time()

5
apps/stable_diffusion/scripts/inpaint.py
View File

@@ -97,7 +97,7 @@ def inpaint_inf(
width,
device,
)
if config_obj != new_config_obj:
if not inpaint_obj or config_obj != new_config_obj:
config_obj = new_config_obj
args.precision = precision
args.batch_size = batch_size
@@ -131,9 +131,6 @@ def inpaint_inf(
args.use_tuned,
)
if not inpaint_obj:
sys.exit("text to image pipeline must not return a null value")
inpaint_obj.scheduler = schedulers[scheduler]
start_time = time.time()

6
apps/stable_diffusion/scripts/txt2img.py
View File

@@ -94,7 +94,7 @@ def txt2img_inf(
width,
device,
)
if config_obj != new_config_obj:
if not txt2img_obj or config_obj != new_config_obj:
config_obj = new_config_obj
args.precision = precision
args.batch_size = batch_size
@@ -105,6 +105,7 @@ def txt2img_inf(
args.iree_vulkan_target_triple = ""
args.use_tuned = True
args.import_mlir = False
args.img_path = ""
set_init_device_flags()
model_id = (
args.hf_model_id
@@ -129,9 +130,6 @@ def txt2img_inf(
low_cpu_mem_usage=args.low_cpu_mem_usage,
)
if not txt2img_obj:
sys.exit("text to image pipeline must not return a null value")
txt2img_obj.scheduler = schedulers[scheduler]
start_time = time.time()