From 37e27b0bd28399b8a9a6324e0c69cfc69c3afa65 Mon Sep 17 00:00:00 2001
From: Daniel Manzke <manzke@users.noreply.github.com>
Date: Wed, 12 Oct 2022 22:38:33 +0200
Subject: [PATCH] Print out the device type which is used

Print out the device type which is used for generating images.
---
 ldm/generate.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ldm/generate.py b/ldm/generate.py
index f9fc364cf3c..d7b40f965fb 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -172,6 +172,7 @@ def __init__(
         # device to Generate(). However the device was then ignored, so
         # it wasn't actually doing anything. This logic could be reinstated.
         device_type = choose_torch_device()
+        print(f'>> Using device_type {device_type}')
         self.device = torch.device(device_type)
         if full_precision:
             if self.precision != 'auto':