We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c82e948 commit 057fc95Copy full SHA for 057fc95
ldm/generate.py
@@ -172,6 +172,7 @@ def __init__(
172
# device to Generate(). However the device was then ignored, so
173
# it wasn't actually doing anything. This logic could be reinstated.
174
device_type = choose_torch_device()
175
+ print(f'>> Using device_type {device_type}')
176
self.device = torch.device(device_type)
177
if full_precision:
178
if self.precision != 'auto':
0 commit comments