Spaces:

ssocean
/

Newborn_Article_Impact_Predict

Running on Zero

ssocean commited on Dec 5, 2024

Commit

5f26632

verified ·

1 Parent(s): f56954d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,6 @@ def init_model():
         model_path,
         num_labels=1,
         load_in_8bit=True,
-        device_map="auto"
     )
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model.eval()

         model_path,
         num_labels=1,
         load_in_8bit=True,
     )
     tokenizer = AutoTokenizer.from_pretrained(model_path)
     model.eval()