So now that I trained it on my face (redoing it with better settings in RunPod) how do I go about ad
So now that I trained it on my face (redoing it with better settings in RunPod) how do I go about adding hero costumes, better backgrounds etc ?




TORCH_USE_CUDA_DSA to enable device-side assertions.TORCH_USE_CUDA_DSA---------------------------------------------------------------------------
KeyError Traceback (most recent call last)
~\AppData\Local\Temp\ipykernel_4440\217437702.py in <module>
1 prompt = "a photo of an astronaut riding a horse on mars"
2 with autocast("cuda"):
----> 3 image = pipe(prompt)["sample"][0]
4
5 image.save("astronaut_rides_horse.png")
~\anaconda3\lib\site-packages\diffusers\utils\outputs.py in __getitem__(self, k)
86 if isinstance(k, str):
87 inner_dict = {k: v for (k, v) in self.items()}
---> 88 return inner_dict[k]
89 else:
90 return self.to_tuple()[k]
KeyError: 'sample'