talphaidze
/

gpt2_201920

Model card Files Files and versions

talphaidze commited on Jun 17, 2025

Commit

7f9fa2c

·

verified ·

1 Parent(s): e10528d

Update modeling.py

Files changed (1) hide show

modeling.py +2 -2

modeling.py CHANGED Viewed

@@ -13,14 +13,14 @@ from huggingface_hub import PyTorchModelHubMixin
 from transformers.utils import ModelOutput
-from .moe import (
     #ExpertChoiceMoE,
     MaskedMoE,
     TimeDependantMoE,
     MoE,
 )
-from .aux_losses import (
     entropy_reg,
     load_balancing_loss,
     router_z_loss,

 from transformers.utils import ModelOutput
+from moe import (
     #ExpertChoiceMoE,
     MaskedMoE,
     TimeDependantMoE,
     MoE,
 )
+from aux_losses import (
     entropy_reg,
     load_balancing_loss,
     router_z_loss,