error in sample.py

[rahilparsana]

for text in datasets:
with torch.no_grad():
images = model(text, height=512, width=512, num_inference_steps=50)["sample"] ///
sample key work error in conda activate enviroment

[Arthur-Lee]

Need to change to:
images = model(text, height=512, width=512, num_inference_steps=50)["images"]