Updated for V2.0

model_inferencing.py

@@ -1,8 +1,6 @@
 TargetModel = None
 def generate_test_image(T2IModel, testPrompt):
-    #prompt = "The quick brown fox jumps over the lazy dog"
     testImage = TargetModel(testPrompt, num_inference_steps=50).images[0]
-    #testImage.save("./image.png")
     
     return testImage
 
@@ -22,13 +20,13 @@ def construct_general_bias_evaluation_prompts(subjects, actions):
             prompts.append('a picture of a ' + subject)
 
     return prompts
-def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageSize):
+def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageWidth, imageHeight):
     guidance = 7.5
     testImages = []
     imageCaptions = [[], []]
     for prompt, ii in zip(prompts, range(len(prompts))):
         testImages+=TargetModel(prompt, num_images_per_prompt=NSamples, num_inference_steps=NSteps,
-                             guidance_scale=guidance, width=imageSize, height=imageSize).images
+                             guidance_scale=guidance, width=imageWidth, height=imageHeight).images
         for nn in range(NSamples):
             imageCaptions[0].append(prompt)                                         # actual prompt used
             imageCaptions[1].append("Prompt: "+str(ii+1)+"    Sample: "+ str(nn+1)) # caption for the image output
@@ -38,13 +36,13 @@ def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageS
     progressBar.empty()
     return (testImages, imageCaptions)
 
-def generate_task_oriented_images(progressBar, barText, prompts, ids, NSamples, NSteps, imageSize):
+def generate_task_oriented_images(progressBar, barText, prompts, ids, NSamples, NSteps, imageWidth, imageHeight):
     guidance = 7.5
     testImages = []
     imageCaptions = [[], []]
     for prompt, jj in zip(prompts, range(len(prompts))):
         testImages+=TargetModel(prompt, num_images_per_prompt=NSamples, num_inference_steps=NSteps,
-                             guidance_scale=guidance, width=imageSize, height=imageSize).images
+                             guidance_scale=guidance, width=imageWidth, height=imageHeight).images
         for nn in range(NSamples):
             imageCaptions[0].append(prompt)                                         # actual prompt used
             imageCaptions[1].append("COCO ID: "+ids[jj]+"    Sample: "+ str(nn+1)) # caption for the image output