Spaces:

schroneko
/

meta-llama-Llama-Guard-3-8B-INT8

Sleeping

schroneko commited on 10 days ago

Commit

2cea941

•

1 Parent(s): 39f6145

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ class LlamaGuardModeration:
         self.model = None
         self.tokenizer = None
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
-        self.model_id = "meta-llama/Llama-Guard-3-8B-INT8"
         self.dtype = torch.bfloat16
         # HuggingFace tokenの取得

         self.model = None
         self.tokenizer = None
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.model_id = "meta-llama/Llama-Guard-3-8B"
         self.dtype = torch.bfloat16
         # HuggingFace tokenの取得