gokaygokay commited on
Commit
c6dea39
1 Parent(s): 4b71d3f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -17,9 +17,8 @@ Fine tuned EleutherAI/pythia-410m using gokaygokay/prompt_description_stable_dif
17
  ```
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
- base_model_name = "gokaygokay/phytia410m_desctoprompt"
21
  model_name = "gokaygokay/phytia410m_desctoprompt"
22
- tokenizer = AutoTokenizer.from_pretrained(base_model_name)
23
  model = AutoModelForCausalLM.from_pretrained(model_name)
24
 
25
  # Your description
 
17
  ```
18
  from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
 
20
  model_name = "gokaygokay/phytia410m_desctoprompt"
21
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
22
  model = AutoModelForCausalLM.from_pretrained(model_name)
23
 
24
  # Your description