Parveshiiii commited on
Commit
050ef42
·
verified ·
1 Parent(s): cd9963e

Update modelling_model.py

Browse files
Files changed (1) hide show
  1. modelling_model.py +4 -1
modelling_model.py CHANGED
@@ -8,7 +8,10 @@ from transformers.modeling_outputs import CausalLMOutputWithPast, BaseModelOutpu
8
  from typing import Optional, Tuple, List, Union
9
  import inspect
10
  from dataclasses import dataclass
11
- from configuration_model import HybridModelConfig
 
 
 
12
 
13
  def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0):
14
  freqs = 1.0 / (theta ** (torch.arange(0, dim, 2)[: (dim // 2)].float() / dim))
 
8
  from typing import Optional, Tuple, List, Union
9
  import inspect
10
  from dataclasses import dataclass
11
+ try:
12
+ from .configuration_model import HybridModelConfig
13
+ except ImportError:
14
+ from configuration_model import HybridModelConfig
15
 
16
  def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0):
17
  freqs = 1.0 / (theta ** (torch.arange(0, dim, 2)[: (dim // 2)].float() / dim))