jvelja commited on
Commit
094cd3d
·
verified ·
1 Parent(s): fb2d798

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. README.md +3 -3
  2. model.safetensors +1 -1
README.md CHANGED
@@ -25,7 +25,7 @@ You can then generate text as follows:
25
  ```python
26
  from transformers import pipeline
27
 
28
- generator = pipeline("text-generation", model="jvelja//tmp/tmpvlxudukz/jvelja/poop_0")
29
  outputs = generator("Hello, my llama is cute")
30
  ```
31
 
@@ -35,8 +35,8 @@ If you want to use the model for training or to obtain the outputs from the valu
35
  from transformers import AutoTokenizer
36
  from trl import AutoModelForCausalLMWithValueHead
37
 
38
- tokenizer = AutoTokenizer.from_pretrained("jvelja//tmp/tmpvlxudukz/jvelja/poop_0")
39
- model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpvlxudukz/jvelja/poop_0")
40
 
41
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
42
  outputs = model(**inputs, labels=inputs["input_ids"])
 
25
  ```python
26
  from transformers import pipeline
27
 
28
+ generator = pipeline("text-generation", model="jvelja//tmp/tmpu21_52sz/jvelja/poop_0")
29
  outputs = generator("Hello, my llama is cute")
30
  ```
31
 
 
35
  from transformers import AutoTokenizer
36
  from trl import AutoModelForCausalLMWithValueHead
37
 
38
+ tokenizer = AutoTokenizer.from_pretrained("jvelja//tmp/tmpu21_52sz/jvelja/poop_0")
39
+ model = AutoModelForCausalLMWithValueHead.from_pretrained("jvelja//tmp/tmpu21_52sz/jvelja/poop_0")
40
 
41
  inputs = tokenizer("Hello, my llama is cute", return_tensors="pt")
42
  outputs = model(**inputs, labels=inputs["input_ids"])
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:13229a73ea99516bc29c5e5701e34f2f3bf825b834b4c9d487721f496a20dc0d
3
  size 497777468
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5071805d15d555f322c62f5a56068f2ff2dbfb40133c07f14cc46742cc565c59
3
  size 497777468