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