xrusnack commited on
Commit
de15af1
·
verified ·
1 Parent(s): 8fabe40

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -2
README.md CHANGED
@@ -13,10 +13,21 @@ language:
13
 
14
  # Uploaded model
15
 
16
- - **Developed by:** xrusnack
17
- - **License:** apache-2.0
18
  - **Finetuned from model :** unsloth/gemma-2-9b-bnb-4bit
19
 
20
  This gemma2 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
 
13
 
14
  # Uploaded model
15
 
 
 
16
  - **Finetuned from model :** unsloth/gemma-2-9b-bnb-4bit
17
 
18
  This gemma2 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
19
+ Example Infernce:
20
+ alpaca_prompt = "### Text: {} ### Summary: {}"
21
+ FastLanguageModel.for_inference(model) # Enable native 2x faster inference
22
+ inputs = tokenizer(
23
+ [
24
+ alpaca_prompt.format(
25
+ "", # text to summarize
26
+ "", # output - leave this blank for generation!
27
+ )
28
+ ], return_tensors = "pt").to("cuda")
29
+
30
+ outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)
31
+ tokenizer.batch_decode(outputs)
32
 
33
  [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)