Changing working device will cause problem

chengzeyi / stable-fast

Best inference performance optimization framework for HuggingFace Diffusers on NVIDIA GPUs.

MIT License

1.19k stars 74 forks source link

Changing working device will cause problem #139

Closed xiyanjoy closed 7 months ago

xiyanjoy commented 7 months ago

stable_fast_test.txt In the presented example, the model is working on the 'cuda' by default. However, if I change the working device to 'cuda:7' as shown in the attach code, there would be some weird problem. error.txt How do I solve this problem?

xiyanjoy commented 7 months ago

Or is there somebody working well in a different device? Look forward to seeing your code!

zhangvia commented 7 months ago

you can use export CUDA_VISIBLE_DEVICES=7 to set device id

xiyanjoy commented 7 months ago

you can use export CUDA_VISIBLE_DEVICES=7 to set device id

It works! Thanks bro.