Harley-ml
/

MNIST-IMG-390k

custom_transformer

Model card Files Files and versions

Harley-ml commited on 23 days ago

Commit

db9b0bd

·

verified ·

1 Parent(s): d929a9a

Update modeling.py

Files changed (1) hide show

modeling.py +1 -1

modeling.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch.nn.functional as F
 from transformers import GenerationMixin, PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutputWithPast
-from configuration import CustomTransformerConfig
 def precompute_rope(head_dim: int, max_seq_len: int, theta: float):

 from transformers import GenerationMixin, PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutputWithPast
+from .configuration import CustomTransformerConfig
 def precompute_rope(head_dim: int, max_seq_len: int, theta: float):