Update app.py
Browse files
app.py
CHANGED
|
@@ -124,6 +124,7 @@ class main():
|
|
| 124 |
self.vae.to(device, dtype=weight_dtype)
|
| 125 |
self.text_encoder.to(device, dtype=weight_dtype)
|
| 126 |
print("")
|
|
|
|
| 127 |
|
| 128 |
self.network = None
|
| 129 |
|
|
|
|
| 124 |
self.vae.to(device, dtype=weight_dtype)
|
| 125 |
self.text_encoder.to(device, dtype=weight_dtype)
|
| 126 |
print("")
|
| 127 |
+
print(self.text_encoder.device)
|
| 128 |
|
| 129 |
self.network = None
|
| 130 |
|