Factor Studios commited on
Commit
e0b5f17
·
verified ·
1 Parent(s): 556bf3a

Update torch_vgpu.py

Browse files
Files changed (1) hide show
  1. torch_vgpu.py +1 -1
torch_vgpu.py CHANGED
@@ -40,7 +40,7 @@ class VGPUDevice:
40
  def __init__(self, vram: Optional[VirtualVRAM] = None):
41
  self.vram = vram or VirtualVRAM()
42
  self.tensor_cores = None # Will be initialized when needed
43
- self.internal_name = "privateuseone" # PyTorch backend name
44
  self.device_name = "vgpu" # User-facing device name
45
  self._register_device()
46
 
 
40
  def __init__(self, vram: Optional[VirtualVRAM] = None):
41
  self.vram = vram or VirtualVRAM()
42
  self.tensor_cores = None # Will be initialized when needed
43
+ self.internal_name = "privateuse1" # PyTorch backend name
44
  self.device_name = "vgpu" # User-facing device name
45
  self._register_device()
46