search

LemurPwned / video-sampler

Effective frame sampling for ML applications.
https://lemurpwned.github.io/video-sampler/
MIT License
16 stars 5 forks source link

Feat/add mps #22

Closed genevera closed 4 months ago

genevera commented 5 months ago

Adds support for MPS on Apple Silicon, updates pytorch check-for-backend syntax, adds a requirements.txt file since I couldn't get it working solely with the pyproject.toml file, bumps python minimum version to 3.10 since we get a TypeError: unsupported operand type(s) for |: 'type' and 'NoneType' error without it.

LemurPwned commented 5 months ago

@genevera Thanks for the contribution! Looks good! If the tests pass and you can help me understand the llama file change, I'm ok with merging.

genevera commented 5 months ago

@genevera Thanks for the contribution! Looks good! If the tests pass and you can help me understand the llama file change, I'm ok with merging.

Thanks! I'm not sure why the test is failing on CUDA. The tests all passed locally for me and I don't have cuda installed. I'll go ahead and switch to is_available to check

LemurPwned commented 4 months ago

Thanks for the contribution!