Factor Studios commited on
Commit
560e47a
·
verified ·
1 Parent(s): e06ef25

Update torch_vgpu.py

Browse files
Files changed (1) hide show
  1. torch_vgpu.py +1 -1
torch_vgpu.py CHANGED
@@ -81,7 +81,7 @@ class VGPUDevice:
81
  def __init__(self, vram: Optional[VirtualVRAM] = None):
82
  self.vram = vram or VirtualVRAM()
83
  self.tensor_cores = None # Will be initialized when needed
84
- self.device_name = "privateuseone:vgpu" # Our registered device type
85
  self._register_device()
86
 
87
  def _register_device(self):
 
81
  def __init__(self, vram: Optional[VirtualVRAM] = None):
82
  self.vram = vram or VirtualVRAM()
83
  self.tensor_cores = None # Will be initialized when needed
84
+ self.device_name = "privateuseone" # Our registered device type
85
  self._register_device()
86
 
87
  def _register_device(self):