MCPcopy
hub / github.com/kohya-ss/sd-scripts / device

Method device

tests/test_custom_offloading_utils.py:44–45  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

42
43 @property
44 def device(self):
45 return next(self.parameters()).device
46
47
48# Device Synchronization Tests

Callers 15

create_upscalerFunction · 0.45
test_cuda_synchronizeFunction · 0.45
test_xpu_synchronizeFunction · 0.45
test_mps_synchronizeFunction · 0.45
test_weights_to_deviceFunction · 0.45
test_swap_weight_devices_cudaFunction · 0.45
test_swap_weight_devices_no_cudaFunction · 0.45
offloaderFunction · 0.45
model_offloaderFunction · 0.45
test_create_backward_hookFunction · 0.45
test_backward_hook_executionFunction · 0.45
test_offloading_integrationFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected