omurberaisik commited on
Commit
e13cfae
·
verified ·
1 Parent(s): 20b6d22

Upload modeling_notoken_gen45.py

Browse files
Files changed (1) hide show
  1. modeling_notoken_gen45.py +6 -1
modeling_notoken_gen45.py CHANGED
@@ -41,7 +41,12 @@ import torch.nn.functional as F
41
  from transformers import PreTrainedModel
42
  from transformers.modeling_outputs import CausalLMOutput
43
 
44
- from configuration_notoken_gen45 import NotokenGen45Config
 
 
 
 
 
45
 
46
 
47
  class RMSNorm(nn.Module):
 
41
  from transformers import PreTrainedModel
42
  from transformers.modeling_outputs import CausalLMOutput
43
 
44
+ try:
45
+ from .configuration_notoken_gen45 import NotokenGen45Config
46
+ except ImportError:
47
+ # Fallback for direct/non-package execution (e.g. running this file
48
+ # standalone rather than through transformers' dynamic module loader).
49
+ from configuration_notoken_gen45 import NotokenGen45Config
50
 
51
 
52
  class RMSNorm(nn.Module):