KLGR123
/

WEPO-gemma-2b

Text Generation

text-generation-inference

Model card Files Files and versions Community

KLGR123 commited on Jul 9, 2024

Commit

56f9db9

·

verified ·

1 Parent(s): 78fec51

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -6,8 +6,8 @@ Below is the reference code for inference. First load the tokenizer and the mode
 ```
 from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained("KLGR123/WEPO-mistral-7b", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("KLGR123/WEPO-mistral-7b", trust_remote_code=True).to('cuda:0')
 ```
 Run a test-demo with random input.

 ```
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("KLGR123/WEPO-gemma-2b", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("KLGR123/WEPO-gemma-2b", trust_remote_code=True).to('cuda:0')
 ```
 Run a test-demo with random input.