jvelja commited on
Commit
1b90c17
1 Parent(s): 1456e84

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/tmpls4jm05k/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
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/tmpls4jm05k/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
40
- model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpls4jm05k/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
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/tmpiicii66o/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
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/tmpiicii66o/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
40
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpiicii66o/jvelja/vllm-gemma2b-stringMatcher-newDataset_1")
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:75beb4b41b4c325c076309bddab5eabb63724e892a9b407697612e6533298240
3
  size 12793376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9352e97dd26b14b659cb0707fbb2458f8fe21d6f82d1d26d735b5b086bbc3742
3
  size 12793376
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3571b911c368250baac46d773bc58faddbb92e72b62bdd301813ed4edbba969e
3
  size 10748
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab889269d78572b5851464731edae861f7467d49e7f7517527bb1e4224ede95d
3
  size 10748