search

TorchMoE / MoE-Infinity

PyTorch library for cost-effective, fast and easy serving of MoE models.
Apache License 2.0
88 stars 5 forks source link

add forward and call #7

Closed future-xy closed 5 months ago

drunkcoding commented 5 months ago

To take any format of argument, this should be

forward(self, *args, **kwargs)
future-xy commented 5 months ago

To take any format of argument, this should be

forward(self, *args, **kwargs)

Fixed in the last commit.