RohitGandikota commited on
Commit
9491756
·
verified ·
1 Parent(s): dfedb72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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