huchenlei / sd-forge-ic-light

A1111/SD Forge extension for IC-Light
Apache License 2.0
198 stars 7 forks source link

Use same Pytorch Device? #8

Closed soten355 closed 6 months ago

soten355 commented 6 months ago

Hello,

Currently the extension is hardcoded to only use CUDA devices.

In the script, scripts>forge_ic_light.py, on line 390, the device is hard coded:

device = torch.device("cuda")

Can you adjust the code to use the same device SD forge is using?