Divyansh008 commited on
Commit
2a670bc
·
verified ·
1 Parent(s): ee721f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -20
README.md CHANGED
@@ -1,20 +1,20 @@
1
- # Merged_Aura_Airavata
2
-
3
- This model merges **Aura_v3_7B** and **Airavata** using upscaling and weighted blending.
4
-
5
- ## Model Details
6
- - **Base Models:** `ResplendentAI/Aura_v3_7B` + `ai4bharat/Airavata`
7
- - **Merging Method:** Upscaling + Weighted Averaging (α=0.5)
8
- - **Tokenizer:** Compatible with `Llama` tokenizers
9
-
10
- ## Usage
11
- ```python
12
- from transformers import AutoModelForCausalLM, AutoTokenizer
13
-
14
- model = AutoModelForCausalLM.from_pretrained("YOUR_USERNAME/Merged_Aura_Airavata")
15
- tokenizer = AutoTokenizer.from_pretrained("YOUR_USERNAME/Merged_Aura_Airavata")
16
-
17
- prompt = "Hello, how are you?"
18
- input_ids = tokenizer(prompt, return_tensors="pt").input_ids
19
- output = model.generate(input_ids)
20
- print(tokenizer.decode(output[0], skip_special_tokens=True))
 
1
+ # Merged_Aura_Airavata
2
+
3
+ This model merges **Aura_v3_7B** and **Airavata** using upscaling and weighted blending.
4
+
5
+ ## Model Details
6
+ - **Base Models:** `ResplendentAI/Aura_v3_7B` + `ai4bharat/Airavata`
7
+ - **Merging Method:** Upscaling + Weighted Averaging (α=0.5)
8
+ - **Tokenizer:** Compatible with `Llama` tokenizers
9
+
10
+ ## Usage
11
+ ```python
12
+ from transformers import AutoModelForCausalLM, AutoTokenizer
13
+
14
+ model = AutoModelForCausalLM.from_pretrained("Divyansh008/Merged_Aura_Airavata")
15
+ tokenizer = AutoTokenizer.from_pretrained("Divyansh008/Merged_Aura_Airavata")
16
+
17
+ prompt = "Hello, how are you?."
18
+ input_ids = tokenizer(prompt, return_tensors="pt").input_ids
19
+ output = model.generate(input_ids)
20
+ print(tokenizer.decode(output[0], skip_special_tokens=True))