masamori
/

gemma-2-2b-finetuned

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

masamori commited on Nov 25, 2024

Commit

5552e44

·

verified ·

1 Parent(s): 1d27f5e

Update ReadME

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ def generate_task_outputs(input_jsonl_path, output_jsonl_path):
         torch_dtype=torch.float16,
         device_map={"": 0}
     )
-    tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
     # LoRAアダプターの適用
     model = PeftModel.from_pretrained(

         torch_dtype=torch.float16,
         device_map={"": 0}
     )
+    tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-2b")
     # LoRAアダプターの適用
     model = PeftModel.from_pretrained(