TechxGenus commited on
Commit
175ade4
1 Parent(s): ab464c8

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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-2b/resolve/main/CodeGemma.jpg">
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-2b")
34
  model = AutoModelForCausalLM.from_pretrained(
35
- "TechxGenus/CodeGemma-2b",
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-2b",
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",