Spaces:
Running
on
Zero
Running
on
Zero
Update model/model_manager.py
Browse files- model/model_manager.py +1 -1
model/model_manager.py
CHANGED
@@ -29,7 +29,7 @@ class ModelManager:
|
|
29 |
pipe = self.loaded_models[model_name]
|
30 |
return pipe
|
31 |
|
32 |
-
def load_guard(self)
|
33 |
model_id = "meta-llama/Meta-Llama-Guard-2-8B"
|
34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
35 |
dtype = torch.bfloat16
|
|
|
29 |
pipe = self.loaded_models[model_name]
|
30 |
return pipe
|
31 |
|
32 |
+
def load_guard(self):
|
33 |
model_id = "meta-llama/Meta-Llama-Guard-2-8B"
|
34 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
35 |
dtype = torch.bfloat16
|