Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,6 @@ df = torch.load(f"{models_path}/files/identity_df.pt")
|
|
91 |
weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
|
92 |
pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=torch.device('cpu')).bfloat16().to(device.value)
|
93 |
|
94 |
-
unet.value, vae.value, text_encoder.value, tokenizer.value, noise_scheduler.value = load_models()
|
95 |
|
96 |
young = gr.State()
|
97 |
young.value = get_direction(df, "Young", pinverse, 1000, device.value)
|
|
|
91 |
weight_dimensions = torch.load(f"{models_path}/files/weight_dimensions.pt")
|
92 |
pinverse = torch.load(f"{models_path}/files/pinverse_1000pc.pt", map_location=torch.device('cpu')).bfloat16().to(device.value)
|
93 |
|
|
|
94 |
|
95 |
young = gr.State()
|
96 |
young.value = get_direction(df, "Young", pinverse, 1000, device.value)
|