Spaces:

poltextlab
/

babelmachine

Running

poltextlab commited on Feb 11

Commit

1341780

verified ·

1 Parent(s): 6b79604

remove implicit model placement

Files changed (1) hide show

interfaces/emotion9.py CHANGED Viewed

@@ -26,7 +26,6 @@ def predict(text, model_id, tokenizer_id):
     device = torch.device("cpu")
     model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", offload_folder="offload", token=HF_TOKEN)
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
-    model.to(device)
     inputs = tokenizer(text,
                        max_length=512,

     device = torch.device("cpu")
     model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", offload_folder="offload", token=HF_TOKEN)
     tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
     inputs = tokenizer(text,
                        max_length=512,