Skip to content

Commit 057fc95

Browse files
manzkelstein
authored andcommitted
Print out the device type which is used
Print out the device type which is used for generating images.
1 parent c82e948 commit 057fc95

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ldm/generate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ def __init__(
172172
# device to Generate(). However the device was then ignored, so
173173
# it wasn't actually doing anything. This logic could be reinstated.
174174
device_type = choose_torch_device()
175+
print(f'>> Using device_type {device_type}')
175176
self.device = torch.device(device_type)
176177
if full_precision:
177178
if self.precision != 'auto':

0 commit comments

Comments
 (0)