Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ def infer(
|
|
73 |
if randomize_seed:
|
74 |
seed = random.randint(0, MAX_SEED)
|
75 |
|
76 |
-
sliderspace_path = f"sliderspace_weights/{slider_space}/slider_{int(discovered_directions[-1])-1}.pt"
|
77 |
|
78 |
for net in networks:
|
79 |
networks[net].load_state_dict(torch.load(sliderspace_path))
|
|
|
73 |
if randomize_seed:
|
74 |
seed = random.randint(0, MAX_SEED)
|
75 |
|
76 |
+
sliderspace_path = f"sliderspace_weights/{slider_space}/slider_{int(discovered_directions.split(' ')[-1])-1}.pt"
|
77 |
|
78 |
for net in networks:
|
79 |
networks[net].load_state_dict(torch.load(sliderspace_path))
|