jvelja commited on
Commit
70d1b62
1 Parent(s): 597ae55

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/tmpdy3yyo0o/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
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/tmpdy3yyo0o/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
40
- model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpdy3yyo0o/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
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/tmpwld9quos/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
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/tmpwld9quos/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
40
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpwld9quos/jvelja/vllm-gemma2b-stringMatcher-newDataset_3")
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:de6d4cc9f6dc5d5948a44c6b39f67980f339707037b6b0c47ee816b93f852950
3
  size 12793376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:83b099c16be4a9a22d71406a8f6e4bb7f5adc2d712dcf57eb022142ee31e744d
3
  size 12793376
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9b5a8134e2c46a346653515cb89cbcdf6a5d33f77f2c949527bd8de0d5fb476a
3
  size 10748
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7fb6725ca03988fb3156a419ad1205110ab3c58ca493649d8e5b92fca0b8155
3
  size 10748