TechxGenus
commited on
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",
|