diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-02 18:20:05 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-02 18:20:05 +0000 |
| commit | 0b94fc5033b8fa5a2613bda21ca0adffbaf73313 (patch) | |
| tree | 696ee21d4778d21740dd033ea6c8b30bf2cc457a /javascript/ui.js | |
| parent | 6365a41f5981efa506dfe4e8fa878b43ca2d8d0c (diff) | |
| parent | a32a068ed13f877560cfdddcfb0b606f476a5728 (diff) | |
| download | stable-diffusion-webui-gfx803-0b94fc5033b8fa5a2613bda21ca0adffbaf73313.tar.gz stable-diffusion-webui-gfx803-0b94fc5033b8fa5a2613bda21ca0adffbaf73313.tar.bz2 stable-diffusion-webui-gfx803-0b94fc5033b8fa5a2613bda21ca0adffbaf73313.zip | |
Merge pull request #1324 from liamkerr/token_updates
Fixing Bugs with Token Counter
Diffstat (limited to 'javascript/ui.js')
| -rw-r--r-- | javascript/ui.js | 25 |
1 files changed, 19 insertions, 6 deletions
diff --git a/javascript/ui.js b/javascript/ui.js index bfe02410..f94ed081 100644 --- a/javascript/ui.js +++ b/javascript/ui.js @@ -199,12 +199,18 @@ let txt2img_textarea, img2img_textarea = undefined; let wait_time = 800 let token_timeout; -function submit_prompt(event, generate_button_id) { - if (event.altKey && event.keyCode === 13) { - event.preventDefault(); - gradioApp().getElementById(generate_button_id).click(); - return; - } +function update_txt2img_tokens(...args) { + update_token_counter("txt2img_token_button") + if (args.length == 2) + return args[0] + return args; +} + +function update_img2img_tokens(...args) { + update_token_counter("img2img_token_button") + if (args.length == 2) + return args[0] + return args; } function update_token_counter(button_id) { @@ -212,3 +218,10 @@ function update_token_counter(button_id) { clearTimeout(token_timeout); token_timeout = setTimeout(() => gradioApp().getElementById(button_id)?.click(), wait_time); } +function submit_prompt(event, generate_button_id) { + if (event.altKey && event.keyCode === 13) { + event.preventDefault(); + gradioApp().getElementById(generate_button_id).click(); + return; + } +}
\ No newline at end of file |