Make the dependency on PyTorch optional

google-ai-edge / model-explorer

A modern model graph visualizer and debugger

Apache License 2.0

1.06k stars 87 forks source link

Make the dependency on PyTorch optional #232

Closed kartynnik closed 3 weeks ago

kartynnik commented 3 weeks ago

The torch dependency brings in 1.7 GB alone, with transitive dependencies adding up to 5 GB. Make it optional to improve the footprint for users interested in visualizing other model formats.