Ma, Guokai 0acf7e9c48 [RFC] add device abstraction to allow other device than CUDA be used (#2221) 1 年之前
..
containers 0acf7e9c48 [RFC] add device abstraction to allow other device than CUDA be used (#2221) 1 年之前
__init__.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
auto_tp.py 2ede0d942a AutoTP Assert Kernel Injection Support (#2939) 1 年之前
inject.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
layers.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
load_checkpoint.py dc01cee5ca using container when loading inference checkpoints (#2875) 1 年之前
module_quantize.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前
policy.py 0acf7e9c48 [RFC] add device abstraction to allow other device than CUDA be used (#2221) 1 年之前
replace_module.py dc01cee5ca using container when loading inference checkpoints (#2875) 1 年之前
replace_policy.py 867da307d0 Inference Refactor (replace_with_policy, model_implementations) (#2554) 1 年之前
utils.py da84e60d98 add missing license info to top of all source code (#2889) 1 年之前