Fixes issue with cuda/current mismatch

This commit is contained in:
Sean McLellan
2022-08-24 13:14:08 -04:00
parent c6b5e930dc
commit b5cdbd3b0b
5 changed files with 21 additions and 12 deletions

View File

@ -191,16 +191,16 @@ def main_loop(t2i,parser,log,infile):
print(f"{newopt.init_img}")
try:
variantResults = t2i.img2img(**vars(newopt))
allVariantResults.append([newopt,variantResults])
except AssertionError as e:
print(e)
continue
allVariantResults.append([newopt,variantResults])
print(f"{opt.variants} Variants generated!")
print("Outputs:")
write_log_message(t2i,opt,results,log)
if len(allVariantResults)>0:
if allVariantResults:
print("Variant outputs:")
for vr in allVariantResults:
write_log_message(t2i,vr[0],vr[1],log)