agadelmoula-avey
/

hf_tutorial

Feature Extraction

Model card Files Files and versions

agadelmoula-avey commited on Oct 15, 2025

Commit

0beb6ff

·

verified ·

1 Parent(s): c0e071f

Update modeling_avey.py

Files changed (1) hide show

modeling_avey.py +1 -1

modeling_avey.py CHANGED Viewed

@@ -12,7 +12,7 @@ from .configuration_avey import AveyConfig
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from torch.utils.checkpoint import checkpoint
-torch._dynamo.config.allow_unspec_int_on_nn_module = True
 class Contextualizer(nn.Module):
     def __init__(self, config: AveyConfig, layer_idx):

 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from torch.utils.checkpoint import checkpoint
+# torch._dynamo.config.allow_unspec_int_on_nn_module = True
 class Contextualizer(nn.Module):
     def __init__(self, config: AveyConfig, layer_idx):