Types & classes — github.com/XueFuzhao/OpenMoE

CodeHub MCPcopy Create free account

hub / github.com/XueFuzhao/OpenMoE / types & classes

Types & classes3 in github.com/XueFuzhao/OpenMoE

⨍ Functions 29 ◇ Types & classes 3

↓ 6 callersClassCapsule

analysis/colossalai_replace/layer.py:272

↓ 1 callersClassStopAfterEosTextGenerated

Logits processor (to use with HuggingFace `generate()` method : https://huggingface.co/docs/transformers/v4.24.0/en/main_classes/ text

script/inference_on_multi_devices.py:8

A class for users to create MoE modules in their models. Args: dim_model (int): Hidden dimension of training model num_experts (i

analysis/colossalai_replace/layer.py:20