diffusers
1f1b6c65 - Device to use (e.g. cpu, cuda:0, cuda:1, etc.) (#1844)

Commit

2 years ago

Device to use (e.g. cpu, cuda:0, cuda:1, etc.) (#1844) * Device to use (e.g. cpu, cuda:0, cuda:1, etc.) * "cuda" if torch.cuda.is_available() else "cpu"

References

#1844 - Device to use (e.g. cpu, cuda:0, cuda:1, etc.)

Author

camenduru

Parents

df2b548e

diffusers 1f1b6c65 - Device to use (e.g. cpu, cuda:0, cuda:1, etc.) (#1844)

diffusers
1f1b6c65 - Device to use (e.g. cpu, cuda:0, cuda:1, etc.) (#1844)