multimodalart HF staff commited on
Commit
d9a8a67
1 Parent(s): af5ea8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
81
 
82
  original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
83
 
84
- def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed, progress=gr.Progress(track_tqdm=True)):
85
  state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
86
  state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
87
  pipe = copy.deepcopy(original_pipe)
 
81
 
82
  original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
83
 
84
+ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1, progress=gr.Progress(track_tqdm=True)):
85
  state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
86
  state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
87
  pipe = copy.deepcopy(original_pipe)