Some test codes do not need to be executed every time

pull/170/head
xingren23 2024-03-08 17:33:25 +08:00
parent 4dc3ceba09
commit 3e2e63028c
6 changed files with 20 additions and 15 deletions

View File

@ -242,5 +242,5 @@ def sort_and_dedupe_csv_file():
# write each item on a new line
fp.write("%s\n" % item)
sort_and_dedupe_csv_file()
if __name__ == "__main__":
sort_and_dedupe_csv_file()

View File

@ -33,4 +33,5 @@ def generateprompts(amount = 1,positive_prompt = "",amountofwords="3"):
print("")
print("All done!")
generateprompts(10,"",5)
if __name__ == "__main__":
generateprompts(10,"",5)

View File

@ -32,4 +32,5 @@ def generateprompts(amount = 1,positive_prompt = "",insanitylevel="0",enhance=Fa
print("")
print("All done!")
generateprompts(1,"photo, photograph, realism",0,False,"", base_model="Stable Cascade")
if __name__ == "__main__":
generateprompts(1,"photo, photograph, realism",0,False,"", base_model="Stable Cascade")

View File

@ -94,7 +94,8 @@ def generateprompts(amount = 1,insanitylevel="5",subject="all", artist="all", im
print("")
print("All done!")
generateprompts(10,5
if __name__ == "__main__":
generateprompts(10,5
,"all" # subject
,"all" # artists
,"all" # image type "only other types", "only templates mode", "art blaster mode", "quality vomit mode", "color cannon mode", "unique art mode", "massive madness mode", "photo fantasy mode", "subject only mode", "fixed styles mode"

View File

@ -36,6 +36,7 @@ def generatepromptvariants(amount = 1,prompt="",insanitylevel="5"):
print("")
print("All done!")
generatepromptvariants(10
if __name__ == "__main__":
generatepromptvariants(10
,"a cute woman" #purple (galaxy) in a (bottle:1.2), <bla:1>, background is a lush jungle and a woman wearing a summer dress, -artmovement-
, 1)

View File

@ -8,13 +8,14 @@ import os
# takes all images with prompt info in their PNG, from the upscale directory, and upscales them
directory = 'C:\\automated_output\\Upscale me\\'
if __name__ == "__main__":
directory = 'C:\\automated_output\\Upscale me\\'
for filename in os.listdir(directory):
f = os.path.join(directory, filename)
# checking if it is a file
if os.path.isfile(f):
img2img1 = call_img2img(f,0.25,1.5,256)
finalfile = call_extras(img2img1)
for filename in os.listdir(directory):
f = os.path.join(directory, filename)
# checking if it is a file
if os.path.isfile(f):
img2img1 = call_img2img(f,0.25,1.5,256)
finalfile = call_extras(img2img1)