Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 =
|
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 |
|