Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AMD card support? #504

Open
agronholm opened this issue Oct 25, 2024 · 5 comments
Open

AMD card support? #504

agronholm opened this issue Oct 25, 2024 · 5 comments

Comments

@agronholm
Copy link

As an owner of a Radeon 7900 XTX, I'm wondering if this project could be made to support AMD cards too. The problem is the xformers dependency which does not support AMD cards. Does Open-Sora-Plan use any nVidia-only features?

@LinB203
Copy link
Member

LinB203 commented Oct 25, 2024

It do not need to install xformers. I think it is ok in AMD card, but I do have this env to test. You can test and report some info here.

@agronholm
Copy link
Author

Why is it declared as a dependency then, here? Also why are all the dependencies pinned to specific versions, rather than the minimum version? If you want to provide a pinned "this worked for us" list of dependencies, then the usual procedure for that would be to provide a requirements.txt which pins the versions of direct and transitive dependencies.

I also kind of wonder why the project needs something like pydub as a dependency, when this project does not even claim to produce audio?

@agronholm
Copy link
Author

It looks like you already have a proper requirements.txt, here. So you don't need to pin exact versions in pyproject.toml too.

@agronholm
Copy link
Author

Digging deeper, it looks like you just dumped everything currently in your virtualenv onto the dependencies list, instead of listing the libraries OpenSora actually needs. I'm trying to untangle the mess, will keep you posted.

@agronholm
Copy link
Author

agronholm commented Oct 26, 2024

From a cursory search, these are the third party libraries actually directly imported by OpenSora:

  • accelerate
  • altair
  • basicsr
  • bs4
  • colorlog
  • decord
  • deepspeed
  • diffusers
  • einops
  • ftfy
  • gdown
  • gradio
  • imageio
  • lcalib (NOT declared as a dependency, NOT on the PyPI, can't even find it on Google!)
  • mmcv (NOT declared as a dependency!)
  • numpy
  • omegaconf
  • opencv-python
  • packaging
  • peft
  • pillow
  • tensorboard
  • timm
  • transformers
  • torch
  • torchvision
  • torch_npu
  • tqdm
  • wandb

Additionally, many of these imports are only needed for training, not inference, so they should be moved to an extra, so as not to make users install dependencies they don't need.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants