jvelja commited on
Commit
5f3e47c
1 Parent(s): 6b32cda

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/tmpsz1t8pzc/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
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/tmpsz1t8pzc/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
40
- model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpsz1t8pzc/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
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/tmp2m1ht0co/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
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/tmp2m1ht0co/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
40
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmp2m1ht0co/jvelja/vllm-gemma2b-stringMatcher-newDataset_0")
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:091a63d95c47d4896f0dab289f033fadceb1dff21638fcb65fd782f3a5f442f5
3
  size 12793376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c486022186a36a078d45dea40dea4bd08720ea5b1ae411ba09dfaa0f2be1047
3
  size 12793376
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b683989d60948db17a83186ada9495a4ad7923e3c304beb5fba9388e3d638607
3
  size 10748
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:69229b3cb870eded86a0ede4c4c5aadc5b91ee552032d2cbc433b86aa75b04b4
3
  size 10748