Spaces:
Running
on
Zero
Running
on
Zero
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 |
|