Update StableDiffusion.vue

Author: AEtheve

electron_app/src/StableDiffusion.vue

@@ -5,6 +5,7 @@
 
 import { send_to_py } from "./py_vue_bridge.js"
 import {get_tokens} from './clip_tokeniser/clip_encoder.js'
+const moment = require('moment')
 
 let notification_sound = new Audio(require('@/assets/notification.mp3'))
 
@@ -144,28 +145,27 @@ export default {
  this.generation_state_msg = iter_time/1000 + " s/it";
  this.times.push(iter_time);
  let median = this.times.sort((a, b) => a - b)[Math.floor(this.times.length / 2)];
- 
- let time_remaining = Math.floor(median*((100-p)*this.nb_its/100) / 1000);
- 
- let minutes = Math.floor(time_remaining / 60);
- let seconds = time_remaining - minutes * 60;
+ let time_remaining = moment.duration(median*((100-p)*this.nb_its/100));
 
  if(time_remaining > 1){ 
- this.remaining_times = `(${minutes.toString().padStart(2, '0')}:${seconds.toString().padStart(2, '0')} left)`;
+ if (time_remaining.hours() > 0) {
+ this.remaining_times = `(${time_remaining.hours()}h${time_remaining.minutes()}m left)`;
+ } else {
+ this.remaining_times = `(${time_remaining.minutes()}m${time_remaining.seconds()}s left)`;
+ }
  if (this.generation_loop) {
  clearInterval(this.generation_loop);
  }
- this.generation_loop = setInterval(() => {
- seconds--;
- if (seconds < 0) {
- seconds = 59;
- minutes--;
+ this.generation_loop = setInterval(() => {
+ time_remaining.subtract(1, 'seconds');
+ if (time_remaining.hours() > 0) {
+ this.remaining_times = `(${time_remaining.hours()}h${time_remaining.minutes()}m left)`;
+ } else {
+ this.remaining_times = `(${time_remaining.minutes()}m${time_remaining.seconds()}s left)`;
  }
- if (minutes == 0 && seconds == 0) {
+ if (time_remaining < 1) {
  clearInterval(this.generation_loop);
  this.remaining_times = "";
- } else{
- this.remaining_times = `(${minutes.toString().padStart(2, '0')}:${seconds.toString().padStart(2, '0')} left)`;
  }
  }, 1000);
  }