elapt1c
/

hrom-testing

Model card Files Files and versions Community

elapt1c commited on 22 days ago

Commit

f67bb08

·

verified ·

1 Parent(s): 2307f6e

Update HROM_Trainer.py

Files changed (1) hide show

HROM_Trainer.py +1 -1

HROM_Trainer.py CHANGED Viewed

@@ -830,7 +830,7 @@ class SafetyManager:
                 return False # Unsafe
         return True # Safe
-    def generate_safely(self, prompt, max_new_tokens=50, temperature=0.7, top_k=50):
         self.model.eval()
         device = next(self.model.parameters()).device

                 return False # Unsafe
         return True # Safe
+    def generate_safely(self, prompt, max_new_tokens=50, temperature=0.5, top_k=50):
         self.model.eval()
         device = next(self.model.parameters()).device