fix lmsd scheduler , cache md5 in filecheck

Author: divamgupta

backends/stable_diffusion/downloader.py

@@ -19,7 +19,19 @@
 import time
 
 def get_md5_file(fpath):
- return str(hashlib.md5(open(fpath, 'rb').read()).hexdigest())
+
+ if os.path.exists(fpath + ".done"):
+ cached = open(fpath + ".done").read()
+ if "_" in cached:
+ md_str = cached.split("_")[0]
+ n_b = int(cached.split("_")[1])
+ if n_b == os.path.getsize(fpath):
+ return md_str
+
+ md_str = str(hashlib.md5(open(fpath, 'rb').read()).hexdigest())
+ n_b = os.path.getsize(fpath)
+ open(fpath + ".done" , "w").write(md_str + "_" + str(n_b))
+ return md_str
 
 
 defualt_downloads_root = os.path.join(projects_root_path, "downloads")

backends/stable_diffusion/schedulers/scheduling_lms_discrete.py

@@ -136,6 +136,8 @@ def set_timesteps(self, num_inference_steps: int):
 
  self.derivatives = []
 
+ self.initial_scale = self.sigmas[0]
+
  self.set_format(tensor_format=self.tensor_format)
 
  def step(

electron_app/src/bridge.js

@@ -16,7 +16,7 @@ function start_bridge() {
  console.log("starting bridge")
  const fs = require('fs')
 
- let script_path = process.env.PY_SCRIPT || "./src/fake_backend.py"; 
+ let script_path = process.env.PY_SCRIPT || "../backends/stable_diffusion/diffusionbee_backend.py"; 
  let bin_path = process.env.BIN_PATH;
  if(bin_path && (fs.existsSync(script_path))){
  python = require('child_process').spawn( bin_path );

electron_app/src/native_functions.js

@@ -463,7 +463,7 @@ function add_custom_pytorch_models(pytorch_model_path, model_name, cb ){
  }
 
 
- let script_path = process.env.PY_SCRIPT || "./src/fake_backend.py"; 
+ let script_path = process.env.PY_SCRIPT || "../backends/stable_diffusion/diffusionbee_backend.py"; 
 
  let out_path = path.join(homedir , ".diffusionbee" , "custom_models" , model_name+".tdict" );
  let proc;