amildravid4292 commited on
Commit
564edc4
·
verified ·
1 Parent(s): d329db4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ class main():
44
  self.pinverse = pinverse
45
 
46
  self.unet, self.vae, self.text_encoder, self.tokenizer, self.noise_scheduler = load_models(device)
 
47
  self.network = None
48
 
49
  young = get_direction(df, "Young", pinverse, 1000, device)
 
44
  self.pinverse = pinverse
45
 
46
  self.unet, self.vae, self.text_encoder, self.tokenizer, self.noise_scheduler = load_models(device)
47
+ print(self.text_encoder.device)
48
  self.network = None
49
 
50
  young = get_direction(df, "Young", pinverse, 1000, device)