Parveshiiii commited on
Commit
cd9963e
·
verified ·
1 Parent(s): 7bbf719

Update modelling_model.py

Browse files
Files changed (1) hide show
  1. modelling_model.py +1 -2
modelling_model.py CHANGED
@@ -8,8 +8,7 @@ from transformers.modeling_outputs import CausalLMOutputWithPast, BaseModelOutpu
8
  from typing import Optional, Tuple, List, Union
9
  import inspect
10
  from dataclasses import dataclass
11
-
12
- #from configuration_model import HybridModelConfig
13
 
14
  def precompute_freqs_cis(dim: int, end: int, theta: float = 10000.0):
15
  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
+ 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))