Update fudge/model.py

fudge/model.py

@@ -6,9 +6,9 @@ import torch.nn.functional as F
 from torch.nn.utils.rnn import pad_sequence, pad_packed_sequence, pack_padded_sequence
 from transformers import AutoTokenizer, AutoModelWithLMHead, pipeline, set_seed, GPT2Tokenizer, GPT2Model, GPT2LMHeadModel, GPT2Config, GPT2ForSequenceClassification, GPT2LMHeadModel, MarianTokenizer
 
-from constants import *
-from util import pad_mask
-from clickbait_classifier import BertClickbaitClassifier, ClickbaitConfig
+from fudge.constants import *
+from fudge.util import pad_mask
+from fudge.clickbait_classifier import BertClickbaitClassifier, ClickbaitConfig
 
 class Model(nn.Module):
     def __init__(self, args, gpt_pad_id, vocab_size, rhyme_group_size=None, glove_embeddings=None, verbose=True):