Mentors4EDU commited on
Commit
011a89f
·
verified ·
1 Parent(s): 4b08de9

Restore relative import in remote code; required by transformers dynamic module loader

Browse files
Files changed (1) hide show
  1. modeling_riemann.py +1 -1
modeling_riemann.py CHANGED
@@ -7,7 +7,7 @@ import torch.nn.functional as F
7
  import torch.utils.checkpoint
8
  from transformers import GenerationMixin, PreTrainedModel
9
  from transformers.modeling_outputs import CausalLMOutputWithPast
10
- from configuration_riemann import RiemannConfig
11
 
12
 
13
  class MultiHeadAttention(nn.Module):
 
7
  import torch.utils.checkpoint
8
  from transformers import GenerationMixin, PreTrainedModel
9
  from transformers.modeling_outputs import CausalLMOutputWithPast
10
+ from .configuration_riemann import RiemannConfig
11
 
12
 
13
  class MultiHeadAttention(nn.Module):