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"]