need detailed tutorial to use Vico with AMD GPU on linux with Rocm

haoosz / ViCo

Official PyTorch codes for the paper: "ViCo: Detail-Preserving Visual Condition for Personalized Text-to-Image Generation"

MIT License

238 stars 15 forks source link

need detailed tutorial to use Vico with AMD GPU on linux with Rocm #6

Open BahzBeih opened 1 year ago

BahzBeih commented 1 year ago

Peace, i need detailed tutorial to use Vico with AMD GPU on linux with Rocm.

haoosz commented 1 year ago

Sorry. I haven't used AMD GPU with Rocm before. I guess the difference lies in the distributed training part, which is well-wrapped by PyTorch Lightning. You may need to check whether PyTorch Lightning supports Rocm.