For our >350M models, here is an example of how to load these within Colab: https://colab.research.google.com/drive/11YU00W-JLNXn-3YckJGOSxFf_TQfCXYr?usp=shari

For our >350M models, here is an example of how to load these within Colab: https://colab.research.google.com/drive/11YU00W-JLNXn-3YckJGOSxFf_TQfCXYr?usp=sharing https://colab.research.google.com/drive/11YU00W-JLNXn-3YckJG... This little hack will force tensor alloc on GPU instead of CPU mem.