Update test.py

test.py

@@ -31,7 +31,7 @@ prompt = "An astronaut riding a green horse"
 widths = [1024, 2048]
 heights = [1024, 2048]
 steps_list = [10, 20]
-batch_sizes = [2]
+batch_sizes = [1,2]
 
 # Warm up
 print("Warming up...")