NousResearch
/

Minos-v1

Text Classification

Model card Files Files and versions Community

Jai Suphavadeeprasit commited on 4 days ago

Commit

8115f10

·

1 Parent(s): 4431d5e

README

Files changed (1) hide show

examples/inference_server.py +1 -1

examples/inference_server.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 import os
 class MinosRefusalClassifier:
-    def __init__(self, model_path_or_name="NousResearch/Minos-v1", use_local=False):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {self.device}")

 import os
 class MinosRefusalClassifier:
+    def __init__(self, model_path_or_name="NousResearch/Minos-v1"):
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         print(f"Using device: {self.device}")