amildravid4292 commited on
Commit
7921611
·
verified ·
1 Parent(s): 945f95c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -79,7 +79,7 @@ class main():
79
  self.weight_dimensions = weight_dimensions
80
  self.pinverse = pinverse
81
 
82
- self.unet, self.vae, self.text_encoder, self.tokenizer, self.noise_scheduler = self.load_models(self.device)
83
  print(self.text_encoder.device)
84
  self.network = None
85
 
 
79
  self.weight_dimensions = weight_dimensions
80
  self.pinverse = pinverse
81
 
82
+ self.unet, self.vae, self.text_encoder, self.tokenizer, self.noise_scheduler = load_models(self.device)
83
  print(self.text_encoder.device)
84
  self.network = None
85