Update upscaler_specialist.py

upscaler_specialist.py

@@ -13,7 +13,9 @@ logger = logging.getLogger(__name__)
 
 class UpscalerSpecialist:
     def __init__(self, device="cuda"):
-        self.device = device if torch.cuda.is_available() else "cpu"
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.cpu_device = torch.device("cpu")
+        self.workspace_dir = workspace_dir
         self.pipe_upsample = None
         self.base_vae = None