/Gpt-Neo-Limited-Vram-Cuda

A notebook that runs GPT-Neo with low vram (6 gb) and cuda acceleration by loading it into gpu memory in smaller parts.

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Stargazers