Update meteor.py

meteor.py

@@ -13,7 +13,7 @@ sample_seed_prefix = b'sample'
 sample_nonce_counter = b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
 
 
-def encode_meteor(model, enc, message, context, finish_sent=False, device='cuda', temp=1.0, precision=16, topk=50000, is_sort=False, randomize_key=False, input_key=sample_key, input_nonce=sample_nonce_counter):
+def encode_meteor(model, enc, message, context, finish_sent=False, device='cpu', temp=1.0, precision=16, topk=50000, is_sort=False, randomize_key=False, input_key=sample_key, input_nonce=sample_nonce_counter):
 
     if randomize_key:
         input_key = os.urandom(64)
@@ -142,7 +142,7 @@ def encode_meteor(model, enc, message, context, finish_sent=False, device='cuda'
 
     return output[len(context):].tolist(), avg_NLL, avg_KL, words_per_bit
 
-def decode_meteor(model, enc, text, context, device='cuda', temp=1.0, precision=16, topk=50000, is_sort=False, input_key=sample_key, input_nonce=sample_nonce_counter):
+def decode_meteor(model, enc, text, context, device='cpu', temp=1.0, precision=16, topk=50000, is_sort=False, input_key=sample_key, input_nonce=sample_nonce_counter):
     # inp is a list of token indices
     # context is a list of token indices
     inp = enc.encode(text)