Update custom.py

custom.py

@@ -11,7 +11,7 @@ import logging
 from torch import Tensor
 import torch
 import torch.nn as nn
-from speechbrain.lobes.models.huggingface_gpt import HuggingFaceGPT
+from speechbrain.lobes.models.huggingface_transformers.gpt import GPT
 try:
     from transformers import GPT2LMHeadModel
     from transformers import GPT2Tokenizer
@@ -23,7 +23,7 @@ except ImportError:
 logger = logging.getLogger(__name__)
 
 
-class HuggingFaceGPT_expanded(HuggingFaceGPT):
+class HuggingFaceGPT_expanded(GPT):
     """This lobe enables the integration of HuggingFace pretrained GPT model.
      Source paper whisper:
         https://life-extension.github.io/2020/05/27/GPT%E6%8A%80%E6%9C%AF%E5%88%9D%E6%8E%A2/language-models.pdf