amildravid4292 commited on
Commit
1d7cba1
·
verified ·
1 Parent(s): f5c27d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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