tests, bug fixes, 1.7.1

Author: divamgupta

backends/model_converter/tdict.py

@@ -123,6 +123,7 @@ def init_read(self):
 
  weights_json_start = extra_head[4]
  weights_json_l = extra_head[5] - extra_head[4]
+ print( "json pos" , weights_json_start , weights_json_l)
 
  self.ctdict_version = int(extra_head[3])
 

backends/stable_diffusion/diffusionbee_backend.py

@@ -243,11 +243,12 @@ def process_opt(d, generator):
 
 def diffusion_bee_main():
 
-
  global p_14 , p_14_np
  download_weights()
+
  print("sdbk mltl Loading Model")
 
+
  def callback(state="" , progress=-1):
  print("sdbk dnpr "+str(progress) )
  if state != "Generating":

backends/stable_diffusion/downloader.py

@@ -104,8 +104,10 @@ def download(self, url, out_fname=None, md5_checksum=None,
  if os.path.exists(out_abs_path.replace(".zip", "")):
  # TODO , actually check if the extracted is valid, but then
  # you might need spereate md5
+ print("sdbk mldn")
  return out_abs_path.replace(".zip", "")
  else:
+ print("sdbk mldn")
  return out_abs_path
 
  with open(out_abs_path, "wb") as f:
@@ -149,5 +151,5 @@ def download(self, url, out_fname=None, md5_checksum=None,
  out_abs_path = out_abs_path.replace(".zip", "")
  dest = shutil.move(extract_path, out_abs_path)
 
-
+ print("sdbk mldn")
  return out_abs_path

backends/stable_diffusion/test_assets/ddd.png

@@ -18,6 +18,146 @@
 
 
 
+def test_1():
+
+ img = sd.generate(
+ prompt="a tree" , 
+ img_height=512, 
+ img_width=512, 
+ seed=1, 
+ tdict_path=None,
+ batch_size=1,
+ dtype=ModelInterface.default_float_type,
+ scheduler='ddim',
+ mode="txt2img" )
+
+ gt_p = "./test_assets/outputs/a_tree_1_ddim.png"
+
+ Image.fromarray(img['img'][0]).show()
+
+
+
+
+
+def test_2():
+
+ img = sd.generate(
+ prompt="a fantasy tiger rightening , Fangs and tusks , massive square jaw with underbite , cybernetic eye , smoking lit cigar , Heavy and crude industrial sci " , 
+ img_height=512, 
+ img_width=512, 
+ seed=145, 
+ tdict_path=None,
+ batch_size=1,
+ dtype=ModelInterface.default_float_type,
+ scheduler='k_euler_ancestral',
+ mode="txt2img" )
+ gt_p = "./test_assets/outputs/tiger_k_euler_ancestral_145.png"
+
+ Image.fromarray(img['img'][0]).show()
+
+
+
+
+
+def test_3():
+
+ img = sd.generate(
+ prompt="a fantasy tiger rightening , Fangs and tusks , massive square jaw with underbite , cybernetic eye , smoking lit cigar , Heavy and crude industrial sci " , 
+ img_height=512, 
+ img_width=512, 
+ seed=145, 
+ tdict_path=None,
+ batch_size=1,
+ dtype=ModelInterface.default_float_type,
+ scheduler='lmsd',
+ mode="txt2img" )
+ gt_p = "./test_assets/outputs/tiger_lmsd_145.png"
+
+ Image.fromarray(img['img'][0]).show()
+
+
+def test_4():
+
+ img = sd.generate(
+ prompt="a fantasy tiger rightening , Fangs and tusks , massive square jaw with underbite , cybernetic eye , smoking lit cigar , Heavy and crude industrial sci " , 
+ img_height=512, 
+ img_width=512, 
+ seed=145, 
+ tdict_path=None,
+ batch_size=1,
+ dtype=ModelInterface.default_float_type,
+ scheduler='pndm',
+ mode="txt2img" )
+
+ gt_p = "./test_assets/outputs/tiger_pndm_145.png"
+
+ Image.fromarray(img['img'][0]).show()
+
+
+
+def test_5():
+
+ inp = "./test_assets/mmm.png"
+ mas = "./test_assets/ddd.png"
+
+ img = sd.generate(
+ prompt="Face of red cat, high resolution, sitting on a park bench" , 
+ img_height=512, 
+ img_width=512, 
+ seed=443136, 
+ input_image=inp, 
+ mask_image=mas, 
+ scheduler='ddim',
+ tdict_path="/Users/divamgupta/.diffusionbee/downloads/sd-v1-5-inpainting_fp16.tdict",
+ mode="inpaint_15" )
+
+ gt_p = "./test_assets/outputs/inp_out_443136.png"
+ Image.fromarray(img['img'][0]).show()
+
+
+
+def test_6():
+
+ inp = "./test_assets/mmm.png"
+ mas = "./test_assets/ddd.png"
+
+ img = sd.generate(
+ prompt="yellow cat, high resolution, sitting on a park bench" , 
+ img_height=512, 
+ img_width=512, 
+ seed=678, 
+ input_image=inp, 
+ mask_image=mas, 
+ scheduler='ddim',
+ tdict_path="/Users/divamgupta/.diffusionbee/downloads/sd-v1-5-inpainting_fp16.tdict",
+ mode="inpaint_15" )
+ gt_p = "./test_assets/outputs/inp_out_678.png"
+ Image.fromarray(img['img'][0]).show()
+
+
+def test_7():
+
+ img = sd.generate(
+ prompt="a tree" , 
+ img_height=512, 
+ img_width=512, 
+ seed=1, 
+ batch_size=2,
+ tdict_path=None,
+ dtype=ModelInterface.default_float_type,
+ scheduler='ddim',
+ mode="txt2img" )
+
+ gt_p = "./test_assets/outputs/a_tree_1_ddim.png"
+ gt_p2 = "./test_assets/outputs/a_tree_1_ddim_2.png"
+
+
+ Image.fromarray(img['img'][0]).show()
+ Image.fromarray(img['img'][1]).show()
+
+
+
+
 
 # inp = "./test_assets/yoga1.jpg"
 
@@ -43,46 +183,43 @@
 
 
 
-inp = "./test_assets/scribble_turtle.png"
-
-
-img = sd.generate(
- prompt="a turtle" , 
- img_height=512, 
- img_width=512, 
- seed=6378, 
- tdict_path=None,
- second_tdict_path="/Users/divamgupta/Downloads/just_control_sd15_scribble_fp16.tdict",
- batch_size=1,
- dtype=ModelInterface.default_float_type,
- scheduler='ddim',
- num_steps=25,
- input_image=inp,
- mode="controlnet" )
-
-
-Image.fromarray(img['img'][0]).show()
+# inp = "./test_assets/scribble_turtle.png"
 
 
+# img = sd.generate(
+# prompt="a turtle" , 
+# img_height=512, 
+# img_width=512, 
+# seed=6378, 
+# tdict_path=None,
+# second_tdict_path="/Users/divamgupta/Downloads/just_control_sd15_scribble_fp16.tdict",
+# batch_size=1,
+# dtype=ModelInterface.default_float_type,
+# scheduler='ddim',
+# num_steps=25,
+# input_image=inp,
+# mode="controlnet" )
 
 
-img = sd.generate(
- prompt="a tortoise" , 
- img_height=512, 
- img_width=512, 
- seed=678, 
- tdict_path=None,
- batch_size=1,
- dtype=ModelInterface.default_float_type,
- scheduler='ddim',
- input_image=inp,
- mode="txt2img" )
+# Image.fromarray(img['img'][0]).show()
 
 
-Image.fromarray(img[0]).show()
+# img = sd.generate(
+# prompt="a tortoise" , 
+# img_height=512, 
+# img_width=512, 
+# seed=678, 
+# tdict_path=None,
+# batch_size=1,
+# dtype=ModelInterface.default_float_type,
+# scheduler='ddim',
+# input_image=inp,
+# mode="txt2img" )
 
 
+# Image.fromarray(img[0]).show()
 
+test_7()
 
 
 exit()

backends/stable_diffusion/test_assets/mmm.png

@@ -1,7 +1,7 @@
 {
  "name": "DiffusionBee",
- "version": "1.7.0",
- "build_number": "0018",
+ "version": "1.7.1",
+ "build_number": "0019",
  "website": "https://diffusionbee.com",
  "description": "Diffusion Bee - Stable Diffusion App.",
  "is_dev": false,