Factor Studios commited on
Commit
d2fdc72
·
verified ·
1 Parent(s): 705a65c

Update torch_vgpu.py

Browse files
Files changed (1) hide show
  1. 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:0")
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