search

substratusai / images

Official Substratus Container Images
1 stars 0 forks source link

only download pytorch model #23

Closed samos123 closed 1 year ago

samos123 commented 1 year ago

Fixes #20

This reduces the model size of many models by 2x or 3x because safetensors and tensorflow model doesn't get used by any of our images. That being said, we might want to consider exposing this as a param in a follow up PR.

brandonjbjelland commented 1 year ago

Nice find eliminating the tensorflow cruft here. 👯