Pythonはじめて触ったもので、どなたか教えてください

Stable Diffusionをcolabで動かしていて、下記コードを使用しています

ここに「縦×横サイズ」「cfg scale」「ステップ数」の指定が出来るようにするにはどのようにコード書けばいいでしょうか？


from torch import autocast
from IPython.display import display
#import matplotlib.pyplot as plt
prompt = ""

for i in range(5):
with autocast("cuda"):
image = pipe(prompt)["sample"][0] # image here is in [PIL format](https://pillow.readthedocs.io/en/stable/)
display(image)