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

bugfix related to the height width params.

Browse Source
This commit is contained in:
Prashant Kumar
2023-01-19 18:47:15 +00:00
parent 7b3d57cff7
commit 9d3af37104
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
shark/examples/shark_inference/stable_diffusion/model_wrappers.py
View File

@@ -50,10 +50,10 @@ def get_input_info(model_info, max_len, width, height):
# for clip, unet and vae respectively.
def get_model_configuration(model_id, max_len, width, height):
if model_id in base_models:
return get_input_info(base_models[model_id], max_len, height, width)
return get_input_info(base_models[model_id], max_len, width, height)
elif model_id in variants:
return get_input_info(
base_models[variants[model_id]], max_len, height, width
base_models[variants[model_id]], max_len, width, height
)
else:
sys.exit(

8
shark/examples/shark_inference/stable_diffusion/resources/base_model.json
View File

@@ -5,8 +5,8 @@
"shape": [
1,
4,
"width",
"height"
"height",
"width"
],
"dtype": "f32"
},
@@ -53,8 +53,8 @@
"shape": [
1,
4,
"width",
"height"
"height",
"width"
],
"dtype": "f32"
},