AjithBharadwaj
/

AB-GPT-Base

Text Generation

Model card Files Files and versions

AjithBharadwaj commited on Jan 9

Commit

5ad39fa

·

verified ·

1 Parent(s): 9973ad8

Update modeling_custom_gpt.py

Files changed (1) hide show

modeling_custom_gpt.py +1 -1

modeling_custom_gpt.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch.nn as nn
 from torch.nn import functional as F
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutputWithPast
-from configuration_custom_gpt import CustomGPTConfig
 # -----------------------------------------------------------------------------
 # 1. Standalone Rotary Embedding (Replacing external dependency)

 from torch.nn import functional as F
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import CausalLMOutputWithPast
+from .configuration_custom_gpt import CustomGPTConfig
 # -----------------------------------------------------------------------------
 # 1. Standalone Rotary Embedding (Replacing external dependency)