examples/community/stable_diffusion_tensorrt_txt2img.py · 05d9baeacd531dc66680d974ec234940e0088d58 · renzhc / diffusers_dcu

Fix TensorRT community pipeline device set function (#3157) · 05d9baea

asfiyab-nvidia authored Apr 21, 2023



pass silence_dtype_warnings as kwarg
Signed-off-by: Asfiya Baig <asfiyab@nvidia.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

05d9baea

stable_diffusion_tensorrt_txt2img.py 36.6 KB

Replace stable_diffusion_tensorrt_txt2img.py