Commit
•
175ade4
1
Parent(s):
ab464c8
Upload README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ license_link: https://ai.google.dev/gemma/terms
|
|
| 10 |
---
|
| 11 |
|
| 12 |
<p align="center">
|
| 13 |
-
<img width="300px" alt="CodeGemma" src="https://huggingface.co/TechxGenus/CodeGemma-
|
| 14 |
</p>
|
| 15 |
|
| 16 |
### CodeGemma
|
|
@@ -30,9 +30,9 @@ PROMPT = """### Instruction
|
|
| 30 |
"""
|
| 31 |
instruction = <Your code instruction here>
|
| 32 |
prompt = PROMPT.format(instruction=instruction)
|
| 33 |
-
tokenizer = AutoTokenizer.from_pretrained("TechxGenus/CodeGemma-
|
| 34 |
model = AutoModelForCausalLM.from_pretrained(
|
| 35 |
-
"TechxGenus/CodeGemma-
|
| 36 |
torch_dtype=torch.bfloat16,
|
| 37 |
device_map="auto",
|
| 38 |
)
|
|
@@ -54,7 +54,7 @@ PROMPT = """<bos>### Instruction
|
|
| 54 |
instruction = <Your code instruction here>
|
| 55 |
prompt = PROMPT.format(instruction=instruction)
|
| 56 |
generator = pipeline(
|
| 57 |
-
model="TechxGenus/CodeGemma-
|
| 58 |
task="text-generation",
|
| 59 |
torch_dtype=torch.bfloat16,
|
| 60 |
device_map="auto",
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
<p align="center">
|
| 13 |
+
<img width="300px" alt="CodeGemma" src="https://huggingface.co/TechxGenus/CodeGemma-7b/resolve/main/CodeGemma.jpg">
|
| 14 |
</p>
|
| 15 |
|
| 16 |
### CodeGemma
|
|
|
|
| 30 |
"""
|
| 31 |
instruction = <Your code instruction here>
|
| 32 |
prompt = PROMPT.format(instruction=instruction)
|
| 33 |
+
tokenizer = AutoTokenizer.from_pretrained("TechxGenus/CodeGemma-7b")
|
| 34 |
model = AutoModelForCausalLM.from_pretrained(
|
| 35 |
+
"TechxGenus/CodeGemma-7b",
|
| 36 |
torch_dtype=torch.bfloat16,
|
| 37 |
device_map="auto",
|
| 38 |
)
|
|
|
|
| 54 |
instruction = <Your code instruction here>
|
| 55 |
prompt = PROMPT.format(instruction=instruction)
|
| 56 |
generator = pipeline(
|
| 57 |
+
model="TechxGenus/CodeGemma-7b",
|
| 58 |
task="text-generation",
|
| 59 |
torch_dtype=torch.bfloat16,
|
| 60 |
device_map="auto",
|