[SD][web] Fix device key error

Signed-Off-by: Gaurav Shukla <[email protected]>

Author: Shukla-Gaurav

web/models/stable_diffusion/cache_objects.py

@@ -91,7 +91,7 @@ def set_models(self, device_key):
             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()

web/models/stable_diffusion/utils.py

@@ -220,7 +220,7 @@ def get_devices_by_name(driver_name):
             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()