minor fix to ldm.dream.args

- If no list of seeds is provided to metadata_dumps(), then the contents of the Args object's seed attribute are used.
afiaka87 · Sep 21, 2022 · ce714a1 · ce714a1
1 parent 48b1253
commit ce714a1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ldm/dream/args.py b/ldm/dream/args.py
@@ -626,6 +626,9 @@ def metadata_dumps(opt,
         rfc_dict['type']  = 'txt2img'
 
     images = []
+    if len(seeds)==0 and opt.seed:
+        seeds=[seed]
+
     for seed in seeds:
         rfc_dict['seed'] = seed
         images.append(copy.copy(rfc_dict))