Spaces:
Runtime error
Runtime error
Factor Studios
commited on
Update torch_vgpu.py
Browse files- 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
|
| 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):
|