alfredplpl
/

gemma-2b-it-ja-poc

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

alfredplpl commited on Mar 5

Commit

8c02750

•

1 Parent(s): 1aeac77

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -21,8 +21,7 @@ tokenizer = AutoTokenizer.from_pretrained(
   "alfredplpl/gemma-2b-it-ja-poc"
 )
 model = AutoModelForCausalLM.from_pretrained(
-  "alfredplpl/gemma-2b-it-ja-poc",
-  torch_dtype=torch.float32
 )
 # プロンプトの準備

   "alfredplpl/gemma-2b-it-ja-poc"
 )
 model = AutoModelForCausalLM.from_pretrained(
+  "alfredplpl/gemma-2b-it-ja-poc"
 )
 # プロンプトの準備