Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -146,7 +146,7 @@ def infer(
|
|
146 |
|
147 |
for net in networks:
|
148 |
networks[net].load_state_dict(torch.load(sliderspace_path))
|
149 |
-
networks[net].set_lora_slider(slider_scale)
|
150 |
with networks[0]:
|
151 |
pass
|
152 |
|
@@ -178,7 +178,7 @@ def infer(
|
|
178 |
sliderspace_path = f"flux_sliderspace_weights/{slider_space}/slider_{int(discovered_directions.split(' ')[-1])-1}.pt"
|
179 |
for net in flux_networks:
|
180 |
flux_networks[net].load_state_dict(torch.load(sliderspace_path))
|
181 |
-
flux_networks[net].set_lora_slider(slider_scale)
|
182 |
with flux_networks[0]:
|
183 |
pass
|
184 |
|
|
|
146 |
|
147 |
for net in networks:
|
148 |
networks[net].load_state_dict(torch.load(sliderspace_path))
|
149 |
+
networks[net].set_lora_slider(-1*slider_scale)
|
150 |
with networks[0]:
|
151 |
pass
|
152 |
|
|
|
178 |
sliderspace_path = f"flux_sliderspace_weights/{slider_space}/slider_{int(discovered_directions.split(' ')[-1])-1}.pt"
|
179 |
for net in flux_networks:
|
180 |
flux_networks[net].load_state_dict(torch.load(sliderspace_path))
|
181 |
+
flux_networks[net].set_lora_slider(-1*slider_scale)
|
182 |
with flux_networks[0]:
|
183 |
pass
|
184 |
|