Spaces:
Runtime error
Runtime error
Factor Studios
commited on
Update torch_vgpu.py
Browse files- torch_vgpu.py +1 -1
torch_vgpu.py
CHANGED
|
@@ -98,7 +98,7 @@ class VGPUDevice:
|
|
| 98 |
raise RuntimeError("VGPU backend not properly initialized")
|
| 99 |
|
| 100 |
# Create device with explicit index - use privateuse1 since that's the actual device type
|
| 101 |
-
self._device = torch.device("privateuse1
|
| 102 |
|
| 103 |
# Store this instance for reuse
|
| 104 |
VGPUDevice._VGPU_INSTANCES[self.device_name] = self
|
|
|
|
| 98 |
raise RuntimeError("VGPU backend not properly initialized")
|
| 99 |
|
| 100 |
# Create device with explicit index - use privateuse1 since that's the actual device type
|
| 101 |
+
self._device = torch.device("privateuse1")
|
| 102 |
|
| 103 |
# Store this instance for reuse
|
| 104 |
VGPUDevice._VGPU_INSTANCES[self.device_name] = self
|