eurus7 commited on
Commit
40180c6
·
1 Parent(s): ee71fe3

Push model using huggingface_hub.

Browse files
Files changed (4) hide show
  1. README.md +42 -0
  2. adapter_config.json +16 -0
  3. adapter_model.bin +3 -0
  4. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - trl
5
+ - transformers
6
+ - reinforcement-learning
7
+ ---
8
+
9
+ # TRL Model
10
+
11
+ This is a [TRL language model](https://github.com/lvwerra/trl) that has been fine-tuned with reinforcement learning to
12
+ guide the model outputs according to a value, function, or human feedback. The model can be used for text generation.
13
+
14
+ ## Usage
15
+
16
+ To use this model for inference, first install the TRL library:
17
+
18
+ ```bash
19
+ python -m pip install trl
20
+ ```
21
+
22
+ You can then generate text as follows:
23
+
24
+ ```python
25
+ from transformers import pipeline
26
+
27
+ generator = pipeline("text-generation", model="eurus7//tmp/tmppczifg03/eurus7/working")
28
+ outputs = generator("Hello, my llama is cute")
29
+ ```
30
+
31
+ If you want to use the model for training or to obtain the outputs from the value head, load the model as follows:
32
+
33
+ ```python
34
+ from transformers import AutoTokenizer
35
+ from trl import AutoModelForCausalLMWithValueHead
36
+
37
+ tokenizer = AutoTokenizer.from_pretrained("eurus7//tmp/tmppczifg03/eurus7/working")
38
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("eurus7//tmp/tmppczifg03/eurus7/working")
39
+
40
+ inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
41
+ outputs = model(**inputs, labels=inputs["input_ids"])
42
+ ```
adapter_config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "base_model_name_or_path": "lvwerra/gpt2-imdb",
3
+ "bias": "none",
4
+ "fan_in_fan_out": true,
5
+ "inference_mode": true,
6
+ "init_lora_weights": true,
7
+ "lora_alpha": 32,
8
+ "lora_dropout": 0.05,
9
+ "modules_to_save": null,
10
+ "peft_type": "LORA",
11
+ "r": 16,
12
+ "target_modules": [
13
+ "c_attn"
14
+ ],
15
+ "task_type": "CAUSAL_LM"
16
+ }
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9000d502320b131c652d85ad69a3aa845bbd99f2136348aed7c54bfe3fec2b5a
3
+ size 2367673
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0b2247ee89ff3ec4430ef3b0c7023589f9fc12a35082eb4beb9eb4ec302ad34
3
+ size 4159