wenhu commited on
Commit
71a2ab3
·
verified ·
1 Parent(s): b19f216

Update model/model_manager.py

Browse files
Files changed (1) hide show
  1. 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