search

salesforce / UniControl

Unified Controllable Visual Generation Model
https://canqin001.github.io/UniControl-Page/
Apache License 2.0
623 stars 35 forks source link

Updated pytorch_lightning import statements #25

Open VikaasVarma opened 1 year ago

VikaasVarma commented 1 year ago

With the newer versions of pytorch the import: from pytorch_lightning.utilities.distributed becomes from pytorch_lightning.utilities.rank_zero

salesforce-cla[bot] commented 1 year ago

Thanks for the contribution! Before we can merge this, we need @VikaasVarma to sign the Salesforce Inc. Contributor License Agreement.