search

jonra1993 / fastapi-alembic-sqlmodel-async

This is a project template which uses FastAPI, Pydantic 2.0, Alembic and async SQLModel as ORM. It shows a complete async CRUD using authentication and role base access control.
MIT License
964 stars 150 forks source link

Switch to Pytorch #68

Closed bazylhorsey closed 1 year ago

bazylhorsey commented 1 year ago

Reasons: nearly all models are supported on pytorch in the Transformer Pipeline feature, much more limited on tensorflow. All that should need changed is the poetry package, all the pipelines should continue to work.

jonra1993 commented 1 year ago

Hello @bazylhorsey it migrates to pytorch in this commit https://github.com/jonra1993/fastapi-alembic-sqlmodel-async/commit/5803ea38ff841e1ae164d282b0f38ed54d2fa314. I decided to use CPU only version to reduce the final container size of computers without gpu.

bazylhorsey commented 1 year ago

That's not true, my understanding is tesnorflow-cpu is actually bigger than tensorflow, because tensorflow has only GPU, tensorflow-cpu has GPU+CPU capabilities meaning it's the bigger of the two. Both PyTorch and tensorflow-cpu are about the same size (1GB) @jonra1993