weirek commited on
Commit
22874d2
·
verified ·
1 Parent(s): 7103c40

Upload model files

Browse files
Files changed (2) hide show
  1. README.md +43 -1
  2. model.safetensors.index.json +2 -2
README.md CHANGED
@@ -1 +1,43 @@
1
- Reinforcement learning aims to bridge the gap between competence and excellence in pre-trained models.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - pytorch
5
+ - causal-lm
6
+ - text-generation
7
+ ---
8
+
9
+ # Affine-2011-5DD686DBbagozdzR7Y94ep2ZJ4oemBQwY2DCZ4mA5QM4s8Qc
10
+
11
+ This model has been fine-tuned for conversational AI tasks.
12
+
13
+ ## Usage
14
+
15
+ ```python
16
+ from transformers import AutoModelForCausalLM, AutoTokenizer
17
+
18
+ model = AutoModelForCausalLM.from_pretrained(
19
+ "weirek/Affine-2011-5DD686DBbagozdzR7Y94ep2ZJ4oemBQwY2DCZ4mA5QM4s8Qc",
20
+ torch_dtype="auto",
21
+ device_map="auto"
22
+ )
23
+ tokenizer = AutoTokenizer.from_pretrained("weirek/Affine-2011-5DD686DBbagozdzR7Y94ep2ZJ4oemBQwY2DCZ4mA5QM4s8Qc")
24
+
25
+ # Generate text
26
+ messages = [
27
+ {"role": "user", "content": "Hello, how are you?"}
28
+ ]
29
+ text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
30
+ inputs = tokenizer(text, return_tensors="pt").to(model.device)
31
+ outputs = model.generate(**inputs, max_new_tokens=256)
32
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
33
+ ```
34
+
35
+ ## Training Details
36
+
37
+ - Training was performed using the Hugging Face Transformers library
38
+ - Model was fine-tuned on conversational data
39
+
40
+ ## Limitations
41
+
42
+ This model inherits limitations from its base model and training data.
43
+ Use responsibly and be aware of potential biases.
model.safetensors.index.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a5c43b9fd27c3ba067265426673431c9cc747d8a5b27e1dab9df9c0fd575e371
3
- size 11396202
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50db0ca60793f0e7d0e89d19e55845a59dc2adff5ea33732f641eb30e09fc802
3
+ size 11396150