Airin-chan commited on
Commit
2b53104
·
verified ·
1 Parent(s): 8cbaa65

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md CHANGED
@@ -78,6 +78,33 @@ How to use Model
78
  print(generate_text(model,input_texts,tokenizer))
79
 
80
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
81
  ```
82
 
83
  Model loss Bechmark:
 
78
  print(generate_text(model,input_texts,tokenizer))
79
 
80
 
81
+ ```
82
+
83
+ How to use Fini_tuned Model
84
+ ----------------------
85
+ ```
86
+ import torch
87
+ from tokenizers import Tokenizer
88
+ from lctlm1 import *
89
+
90
+ from tokenizers import Tokenizer
91
+ tokenizer = Tokenizer.from_file("tokenizer.json")
92
+ model = LMLCT1()
93
+ model.load_state_dict(torch.load("lctlm1_finetuned.pth"))
94
+
95
+ if __name__ == "__main__" :
96
+
97
+ state_stop = False
98
+
99
+ while (state_stop == False) :
100
+ users = input("type input here : ")
101
+ response = generate_tesk(model=model,texts=users,tokenizer=tokenizer,temperature=1.0)
102
+ print(f"response : \n {response}")
103
+ stop = input("stop ? ")
104
+ if stop == "y" or stop == "Y" :
105
+ break
106
+
107
+
108
  ```
109
 
110
  Model loss Bechmark: