owllwo98 commited on
Commit
03f66c3
·
verified ·
1 Parent(s): f26ba6a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +25 -46
README.md CHANGED
@@ -1,58 +1,37 @@
1
  ---
2
  base_model: google/functiongemma-270m-it
3
- library_name: transformers
4
- model_name: mobile-actions-functiongemma
5
  tags:
6
- - generated_from_trainer
7
- - trl
8
- - sft
9
- licence: license
 
 
 
10
  ---
11
 
12
- # Model Card for mobile-actions-functiongemma
13
 
14
- This model is a fine-tuned version of [google/functiongemma-270m-it](https://huggingface.co/google/functiongemma-270m-it).
15
- It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
- ## Quick start
18
-
19
- ```python
20
- from transformers import pipeline
21
-
22
- question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
- generator = pipeline("text-generation", model="None", device="cuda")
24
- output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
- print(output["generated_text"])
26
- ```
27
-
28
- ## Training procedure
29
-
30
-
31
 
 
 
 
 
32
 
33
- This model was trained with SFT.
34
-
35
- ### Framework versions
36
-
37
- - TRL: 0.25.1
38
- - Transformers: 4.57.1
39
- - Pytorch: 2.9.0+cu126
40
- - Datasets: 4.4.1
41
- - Tokenizers: 0.22.1
42
-
43
- ## Citations
44
 
 
 
 
 
45
 
 
 
 
46
 
47
- Cite TRL as:
48
-
49
- ```bibtex
50
- @misc{vonwerra2022trl,
51
- title = {{TRL: Transformer Reinforcement Learning}},
52
- author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
53
- year = 2020,
54
- journal = {GitHub repository},
55
- publisher = {GitHub},
56
- howpublished = {\url{https://github.com/huggingface/trl}}
57
- }
58
- ```
 
1
  ---
2
  base_model: google/functiongemma-270m-it
 
 
3
  tags:
4
+ - function-calling
5
+ - mobile-actions
6
+ - gemma
7
+ - fine-tuned
8
+ license: gemma
9
+ language:
10
+ - en
11
  ---
12
 
13
+ # FunctionGemma 270M - Mobile Actions
14
 
15
+ Fine-tuned model based on `google/functiongemma-270m-it` for mobile action function calling.
 
16
 
17
+ ## Functions
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
+ This model can call the following functions:
20
+ - `change_background_color`: Changes app background color
21
+ - `search_web`: Searches the web for information
22
+ - `set_timer`: Sets a timer for specified duration
23
 
24
+ ## Training
 
 
 
 
 
 
 
 
 
 
25
 
26
+ - **Base Model**: google/functiongemma-270m-it
27
+ - **Training Data**: 30 custom examples (27 train, 3 eval)
28
+ - **Method**: LoRA fine-tuning
29
+ - **Epochs**: 2
30
 
31
+ ## Usage
32
+ ```python
33
+ from transformers import AutoModelForCausalLM, AutoTokenizer
34
 
35
+ model = AutoModelForCausalLM.from_pretrained("owllwo98/functiongemma-270m-it-mobile-actions")
36
+ tokenizer = AutoTokenizer.from_pretrained("owllwo98/functiongemma-270m-it-mobile-actions")
37
+ ```