prithivMLmods commited on
Commit
916d78e
·
verified ·
1 Parent(s): d234e80

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +112 -1
README.md CHANGED
@@ -1,4 +1,115 @@
1
  ---
2
  license: creativeml-openrail-m
 
 
 
 
 
 
 
 
 
3
  ---
4
- ![gwq2.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/Ayc6YKE6FKYKb8Mible4z.png)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: creativeml-openrail-m
3
+ language:
4
+ - en
5
+ base_model:
6
+ - prithivMLmods/GWQ-9B-Preview
7
+ pipeline_tag: text-generation
8
+ library_name: transformers
9
+ tags:
10
+ - gemma2
11
+ - text-generation-inference
12
  ---
13
+ ![gwq2.png](https://cdn-uploads.huggingface.co/production/uploads/65bb837dbfb878f46c77de4c/Ayc6YKE6FKYKb8Mible4z.png)
14
+
15
+ # **GWQ-9B-Preview2**
16
+
17
+ GWQ - Gemma with Questions Prev is a family of lightweight, state-of-the-art open models from Google, built using the same research and technology employed to create the Gemini models. These models are text-to-text, decoder-only large language models, available in English, with open weights for both pre-trained and instruction-tuned variants. Gemma models are well-suited for a variety of text generation tasks, including question answering, summarization, and reasoning. GWQ is fine-tuned on the Chain of Continuous Thought Synthetic Dataset, built upon the Gemma2forCasualLM architecture.
18
+
19
+ # **Running GWQ Demo**
20
+
21
+ ```python
22
+ # pip install accelerate
23
+ from transformers import AutoTokenizer, AutoModelForCausalLM
24
+ import torch
25
+
26
+ tokenizer = AutoTokenizer.from_pretrained("prithivMLmods/GWQ-9B-Preview2")
27
+ model = AutoModelForCausalLM.from_pretrained(
28
+ "prithivMLmods/GWQ-9B-Preview2",
29
+ device_map="auto",
30
+ torch_dtype=torch.bfloat16,
31
+ )
32
+
33
+ input_text = "Write me a poem about Machine Learning."
34
+ input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
35
+
36
+ outputs = model.generate(**input_ids, max_new_tokens=32)
37
+ print(tokenizer.decode(outputs[0]))
38
+ ```
39
+
40
+ You can ensure the correct chat template is applied by using `tokenizer.apply_chat_template` as follows:
41
+ ```python
42
+ messages = [
43
+ {"role": "user", "content": "Write me a poem about Machine Learning."},
44
+ ]
45
+ input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt", return_dict=True).to("cuda")
46
+
47
+ outputs = model.generate(**input_ids, max_new_tokens=256)
48
+ print(tokenizer.decode(outputs[0]))
49
+ ```
50
+
51
+ # **Key Architecture**
52
+
53
+ 1. **Transformer-Based Design**:
54
+ Gemma 2 leverages the transformer architecture, utilizing self-attention mechanisms to process input text and capture contextual relationships effectively.
55
+
56
+ 2. **Lightweight and Efficient**:
57
+ It is designed to be computationally efficient, with fewer parameters compared to larger models, making it ideal for deployment on resource-constrained devices or environments.
58
+
59
+ 3. **Modular Layers**:
60
+ The architecture consists of modular encoder and decoder layers, allowing flexibility in adapting the model for specific tasks like text generation, summarization, or classification.
61
+
62
+ 4. **Attention Mechanisms**:
63
+ Gemma 2 employs multi-head self-attention to focus on relevant parts of the input text, improving its ability to handle long-range dependencies and complex language structures.
64
+
65
+ 5. **Pre-training and Fine-Tuning**:
66
+ The model is pre-trained on large text corpora and can be fine-tuned for specific tasks, such as markdown processing in ReadM.Md, to enhance its performance on domain-specific data.
67
+
68
+ 6. **Scalability**:
69
+ The architecture supports scaling up or down based on the application's requirements, balancing performance and resource usage.
70
+
71
+ 7. **Open-Source and Customizable**:
72
+ Being open-source, Gemma 2 allows developers to modify and extend its architecture to suit specific use cases, such as integrating it into tools like ReadM.Md for markdown-related tasks.
73
+
74
+ # **Intended Use of GWQ2 (Gemma with Questions2)**
75
+
76
+ 1. **Question Answering:**
77
+ The model excels in generating concise and relevant answers to user-provided queries across various domains.
78
+
79
+ 2. **Summarization:**
80
+ It can be used to summarize large bodies of text, making it suitable for news aggregation, academic research, and report generation.
81
+
82
+ 3. **Reasoning Tasks:**
83
+ GWQ is fine-tuned on the Chain of Continuous Thought Synthetic Dataset, which enhances its ability to perform reasoning, multi-step problem solving, and logical inferences.
84
+
85
+ 4. **Text Generation:**
86
+ The model is ideal for creative writing tasks such as generating poems, stories, and essays. It can also be used for generating code comments, documentation, and markdown files.
87
+
88
+ 5. **Instruction Following:**
89
+ GWQ’s instruction-tuned variant is suitable for generating responses based on user instructions, making it useful for virtual assistants, tutoring systems, and automated customer support.
90
+
91
+ 6. **Domain-Specific Applications:**
92
+ Thanks to its modular design and open-source nature, the model can be fine-tuned for specific tasks like legal document summarization, medical record analysis, or financial report generation.
93
+
94
+ ## **Limitations of GWQ2**
95
+
96
+ 1. **Resource Requirements:**
97
+ Although lightweight compared to larger models, the 9B parameter size still requires significant computational resources, including GPUs with large memory for inference.
98
+
99
+ 2. **Knowledge Cutoff:**
100
+ The model’s pre-training data may not include recent information, making it less effective for answering queries on current events or newly developed topics.
101
+
102
+ 3. **Bias in Outputs:**
103
+ Since the model is trained on publicly available datasets, it may inherit biases present in those datasets, leading to potentially biased or harmful outputs in sensitive contexts.
104
+
105
+ 4. **Hallucinations:**
106
+ Like other large language models, GWQ can occasionally generate incorrect or nonsensical information, especially when asked for facts or reasoning outside its training scope.
107
+
108
+ 5. **Lack of Common-Sense Reasoning:**
109
+ While GWQ is fine-tuned for reasoning, it may still struggle with tasks requiring deep common-sense knowledge or nuanced understanding of human behavior and emotions.
110
+
111
+ 6. **Dependency on Fine-Tuning:**
112
+ For optimal performance on domain-specific tasks, fine-tuning on relevant datasets is required, which demands additional computational resources and expertise.
113
+
114
+ 7. **Context Length Limitation:**
115
+ The model’s ability to process long documents is limited by its maximum context window size. If the input exceeds this limit, truncation may lead to loss of important information.