From ebc2a3cdc9187dc4c778d6812926856fef6cdef6 Mon Sep 17 00:00:00 2001 From: Daniel Hesslow Date: Thu, 25 May 2023 08:11:43 +0000 Subject: [PATCH] Create README.md --- README.md | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 README.md diff --git a/README.md b/README.md new file mode 100644 index 0000000..a9e8fdd --- /dev/null +++ b/README.md @@ -0,0 +1,33 @@ + +# Example usage + +```python +from transformers import AutoTokenizer, AutoModelForCausalLM +import transformers +import torch + +model = "tiiuae/falcon-40B" + +tokenizer = AutoTokenizer.from_pretrained(model) +pipeline = transformers.pipeline( + "text-generation", + model=model, + tokenizer=tokenizer, + torch_dtype=torch.bfloat16, + trust_remote_code=True, + device_map="auto", +) +sequences = pipeline( + "My favourite data joke is", + max_length=200, + do_sample=True, + top_k=10, + num_return_sequences=1, + eos_token_id=tokenizer.eos_token_id, +) +print("=" * 30) +print("Results:") +for seq in sequences: + print(f"Result: {seq['generated_text']}") + +```