Upload 2 files

app.py

@@ -4,6 +4,7 @@ import torch
 
 from better_transformer import *
 
+
 def main():
 
     # Enable CUDA if available and load in tokenizer

better_transformer.py

@@ -301,7 +301,6 @@ def load_tokenizer(device):
 
 def load_big_model(tokenizer, device):
     ## Model architecture
-    set_seed(42)
     N_HEAD = 16
     N_LAYER = 8
     N_EMBD = 768