Spaces:

Nadil-K
/

lm-spell

Sleeping

Nadil Karunarathna commited on Jun 4

Commit

274faae

1 Parent(s): 6baca04

remove threads

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ def init():
     hf_token = os.environ.get("HF_TOKEN")
     model = MT5ForConditionalGeneration.from_pretrained("lm-spell/mt5-base-ft-ssc", token=hf_token)
-    torch.set_num_threads(16)
     tokenizer = T5TokenizerFast.from_pretrained("google/mt5-base")
     tokenizer.add_special_tokens({'additional_special_tokens': ['<ZWJ>']})

     hf_token = os.environ.get("HF_TOKEN")
     model = MT5ForConditionalGeneration.from_pretrained("lm-spell/mt5-base-ft-ssc", token=hf_token)
+    # torch.set_num_threads(16)
     tokenizer = T5TokenizerFast.from_pretrained("google/mt5-base")
     tokenizer.add_special_tokens({'additional_special_tokens': ['<ZWJ>']})