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