dfurman
/

Llama-2-13B-Instruct-v0.2

Text Generation

Model card Files Files and versions Community

dfurman commited on Jul 27, 2023

Commit

5d17f6b

·

1 Parent(s): 51f4d06

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -153,7 +153,7 @@ def llama_generate(
     tokenizer: AutoTokenizer,
     prompt: str,
     max_new_tokens: int = 128,
-    temperature: int = 1.0,
 ) -> str:
     """
     Initialize the pipeline

     tokenizer: AutoTokenizer,
     prompt: str,
     max_new_tokens: int = 128,
+    temperature: float = 0.92,
 ) -> str:
     """
     Initialize the pipeline