general-preference
/

GPM-Gemma-2B

Model card Files Files and versions Community

kirigayahitsugi commited on Oct 14

Commit

a2cadc2

•

1 Parent(s): f2ebf3a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -164,7 +164,7 @@ class GPMPipeline:
         self.truncation = truncation
         self.max_length = max_length
         self.padding = padding
-        self.tau = 0.1
         config = AutoConfig.from_pretrained(model_name_or_path, trust_remote_code=True)
         config._attn_implementation = "flash_attention_2"

         self.truncation = truncation
         self.max_length = max_length
         self.padding = padding
+        self.tau = tau
         config = AutoConfig.from_pretrained(model_name_or_path, trust_remote_code=True)
         config._attn_implementation = "flash_attention_2"