Making large AI models cheaper, faster and more accessible
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
HELSON e5ea3fdeef
[gemini] add GeminiMemoryManger (#832)
3 years ago
..
amp
builder modefied the pp build for ckpt adaptation (#803) 3 years ago
cli [cli] added check installation cli 3 years ago
communication
context
engine
gemini [gemini] add GeminiMemoryManger (#832) 3 years ago
kernel
logging
nn [gemini] add GeminiMemoryManger (#832) 3 years ago
registry [dependency] removed torchvision (#833) 3 years ago
tensor [hotfix] the bug of numel() in ColoTensor (#845) 3 years ago
testing
trainer
utils [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 3 years ago
zero [gemini] add GeminiMemoryManger (#832) 3 years ago
__init__.py
constants.py
core.py
global_variables.py
initialize.py modefied the pp build for ckpt adaptation (#803) 3 years ago