Spaces:
Runtime error
Runtime error
Dusan Svilarkovic
commited on
Commit
•
706cd84
1
Parent(s):
4da12b6
Fix
Browse files
script.py
CHANGED
@@ -46,7 +46,8 @@ model_args.task = 'clickbait'
|
|
46 |
model_args.device = device
|
47 |
model_args.checkpoint = '../checkpoint-1464/'
|
48 |
|
49 |
-
conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
|
|
50 |
conditioning_model = conditioning_model.to(device)
|
51 |
conditioning_model.eval()
|
52 |
|
|
|
46 |
model_args.device = device
|
47 |
model_args.checkpoint = '../checkpoint-1464/'
|
48 |
|
49 |
+
# conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
50 |
+
conditioning_model = Model(model_args, pad_id, vocab_size=None) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
|
51 |
conditioning_model = conditioning_model.to(device)
|
52 |
conditioning_model.eval()
|
53 |
|