Resolving Env Differences & Installing GPU PyTorch for FitDiT

#4
by onion1234567 - opened

image.png

image.png

I’ve noticed that the environment specifications on GitHub and on Hugging Face are different:
How should I reconcile these two sets of environment values? And if I want to run the model on GPU, I understand I need to install a different build of PyTorch— which exact environment (CUDA version, PyTorch package, etc.) should I install to enable GPU support?

I’m really enjoying using this feature—thanks so much!

Sign up or log in to comment