Abhaykoul commited on
Commit
265ccd5
·
1 Parent(s): f34456b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -1
README.md CHANGED
@@ -1,3 +1,58 @@
1
  ---
 
 
 
 
 
 
 
 
 
 
2
  inference: false
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: mit
3
+ datasets:
4
+ - HuggingFaceH4/ultrachat_200k
5
+ - openchat/openchat_sharegpt4_dataset
6
+ - Open-Orca/SlimOrca
7
+ language:
8
+ - en
9
+ pipeline_tag: conversational
10
+ tags:
11
+ - text-generation-inference
12
  inference: false
13
+ ---
14
+ # HelpingAI-Lite-chat
15
+
16
+ **HelpingAI-Lite-chat is a conversational model with 1 billion parameters.**
17
+ It is finetuned from HelpingAI and falcon
18
+
19
+ **🎯 Purpose**
20
+
21
+ The HelpingAI-Lite-chat aims to add conversational capabilities to the HelpingAI-Lite model. This initiative is driven by the need for a smaller, open-source, instruction-finetuned, ready-to-use model, suitable for users with limited computational resources, like lower-end consumer GPUs.
22
+
23
+ ## 📖 Example Code
24
+
25
+ ```python
26
+ from transformers import AutoTokenizer, AutoModelForCausalLM
27
+ import torch
28
+ model_name = "OEvortex/HelpingAI-Lite-chat"
29
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
30
+ model = AutoModelForCausalLM.from_pretrained(
31
+ model_name, device_map="auto", torch_dtype=torch.bfloat16
32
+ )
33
+ chat_history = [
34
+ {"role": "user", "content": "Hello!"},
35
+ {"role": "assistant", "content": "Hello! How can I assist you today?"},
36
+ {"role": "user", "content": "Explain what AI is."},
37
+ ]
38
+ input_ids = tokenizer.apply_chat_template(
39
+ chat_history, tokenize=True, add_generation_prompt=True, return_tensors="pt"
40
+ ).to(model.device)
41
+ output_tokens = model.generate(
42
+ input_ids,
43
+ do_sample=True,
44
+ temperature=0.7,
45
+ repetition_penalty=1.05,
46
+ max_new_tokens=200,
47
+ )
48
+ output_text = tokenizer.decode(
49
+ output_tokens[0][len(input_ids[0]) :], skip_special_tokens=True
50
+ )
51
+ print(output_text)
52
+ ```
53
+
54
+ ## ⚠️ Limitations
55
+
56
+ This model may generate inaccurate or misleading information and is prone to hallucination, creating plausible but false narratives. It lacks the ability to discern factual content from fiction and may inadvertently produce biased, harmful or offensive content. Its understanding of complex, nuanced queries is limited. Users should be aware of this and verify any information obtained from the model.
57
+
58
+ The model is provided 'as is' without any warranties, and the creators are not liable for any damages arising from its use. Users are responsible for their interactions with the model.