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][web] Fix device key error

Browse Source 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
This commit is contained in:
Gaurav Shukla
2023-01-10 20:43:22 +05:30
parent f94a571773
commit 472185c3e4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web/models/stable_diffusion/cache_objects.py
View File

@@ -91,7 +91,7 @@ class ModelCache:
self.device = device_key
self.variant = args.variant
self.version = args.version
args.device = device_key.split("(", 1)[1].strip()[:-1]
args.device = device_key.split("=>", 1)[1].strip()
args.max_length = 64
args.use_tuned = True
set_init_device_flags()

2
web/models/stable_diffusion/utils.py
View File

@@ -220,7 +220,7 @@ def get_available_devices():
print(f"{driver_name} devices are not available.")
else:
for i, device in enumerate(device_list_dict):
device_list.append(f"{device['name']} ({driver_name}://{i})")
device_list.append(f"{device['name']} => {driver_name}://{i}")
return device_list
set_iree_runtime_flags()