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] Fix multiple call to device check (#1007)

Browse Source 
- Also makes the dark theme default.
- Fix custom_vae parameter in img2img.

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2023-02-14 01:27:52 +05:30
committed by GitHub
parent 87fd13d8eb
commit dd2e482214
6 changed files with 50 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -219,6 +219,7 @@ def img2img_inf(
args.import_mlir,
args.hf_model_id,
args.ckpt_loc,
args.custom_vae,
args.precision,
args.max_length,
args.batch_size,
@@ -288,6 +289,7 @@ if __name__ == "__main__":
args.import_mlir,
args.hf_model_id,
args.ckpt_loc,
args.custom_vae,
args.precision,
args.max_length,
args.batch_size,

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

@@ -220,6 +220,7 @@ def inpaint_inf(
args.import_mlir,
args.hf_model_id,
args.ckpt_loc,
args.custom_vae,
args.precision,
args.max_length,
args.batch_size,
@@ -301,6 +302,7 @@ if __name__ == "__main__":
args.import_mlir,
args.hf_model_id,
args.ckpt_loc,
args.custom_vae,
args.precision,
args.max_length,
args.batch_size,