jvelja commited on
Commit
32aa4f9
1 Parent(s): f2e5162

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. adapter_model.safetensors +1 -1
  3. pytorch_model.bin +1 -1
README.md CHANGED
@@ -26,7 +26,7 @@ You can then generate text as follows:
26
  ```python
27
  from transformers import pipeline
28
 
29
- generator = pipeline("text-generation", model="jvelja//tmp/tmpldyi1a4h/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
30
  outputs = generator("Hello, my llama is cute")
31
  ```
32
 
@@ -36,8 +36,8 @@ If you want to use the model for training or to obtain the outputs from the valu
36
  from transformers import AutoTokenizer
37
  from trl import AutoModelForCausalLMWithValueHead
38
 
39
- tokenizer = AutoTokenizer.from_pretrained("jvelja//tmp/tmpldyi1a4h/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
40
- model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpldyi1a4h/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
41
 
42
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
43
  outputs = model(**inputs, labels=inputs["input_ids"])
 
26
  ```python
27
  from transformers import pipeline
28
 
29
+ generator = pipeline("text-generation", model="jvelja//tmp/tmpfyuwszc5/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
30
  outputs = generator("Hello, my llama is cute")
31
  ```
32
 
 
36
  from transformers import AutoTokenizer
37
  from trl import AutoModelForCausalLMWithValueHead
38
 
39
+ tokenizer = AutoTokenizer.from_pretrained("jvelja//tmp/tmpfyuwszc5/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
40
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpfyuwszc5/jvelja/vllm-gemma2b-stringMatcher-newDataset_4")
41
 
42
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
43
  outputs = model(**inputs, labels=inputs["input_ids"])
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ad206bc329727c6b1f445107392a42a30fd8cda94ea6965530159112e7f66020
3
  size 12793376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a6e6d0fc8618681d22317931190cc2ca16ca92e0fdd568fec687bda04c0d36e
3
  size 12793376
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3764f2f8b4b667f5c8476d489d6c21a271a6540c8fc5419b3d06bd83002e359c
3
  size 10748
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4f4c99b6cc1a600fc0d8187a72e86ce3c12bc3880849b9dcc0d90a2ed67fde5
3
  size 10748