Abdurrahmanesc commited on
Commit
707524c
·
verified ·
1 Parent(s): 06496e9

Update modeling_tinygpt.py

Browse files
Files changed (1) hide show
  1. modeling_tinygpt.py +2 -1
modeling_tinygpt.py CHANGED
@@ -4,7 +4,8 @@ import torch.nn.functional as F
4
  from torch.nn import CrossEntropyLoss
5
  from transformers.modeling_outputs import CausalLMOutput
6
  from transformers.modeling_utils import PreTrainedModel
7
- from configuration_tinygpt import TinyGPTConfig # Changed from relative to absolute import
 
8
 
9
  # -------------------------
10
  # TinyGPTConfig (Required)
 
4
  from torch.nn import CrossEntropyLoss
5
  from transformers.modeling_outputs import CausalLMOutput
6
  from transformers.modeling_utils import PreTrainedModel
7
+
8
+ from .configuration_tinygpt import TinyGPTConfig #to look inside the folder
9
 
10
  # -------------------------
11
  # TinyGPTConfig (Required)