src/diffusers/loaders/single_file_utils.py · 6bceaea3fe5937c4ea5e833a402b24492860d0f7 · renzhc / diffusers_dcu · GitLab

Find file Blame History Permalink

[from_single_file] pass `torch_dtype` to `set_module_tensor_to_device` (#6994) · 779eef95
YiYi Xu authored Feb 18, 2024
```
fix
Co-authored-by: yiyixuxu <yixu310@gmail,com>
```
779eef95

single_file_utils.py 59 KB