Yuchan
commited on
Update Mo.py
Browse files
Mo.py
CHANGED
|
@@ -276,7 +276,7 @@ history = model.fit(dataset, epochs=1, verbose=1)
|
|
| 276 |
model.save_weights("model.weights.h5")
|
| 277 |
print("모델 가중치 저장 완료!")
|
| 278 |
|
| 279 |
-
def generate_text_topp(model, prompt, max_len=
|
| 280 |
model_input = text_to_ids(f"<start> {prompt}")
|
| 281 |
model_input = model_input[:max_len]
|
| 282 |
generated = list(model_input)
|
|
|
|
| 276 |
model.save_weights("model.weights.h5")
|
| 277 |
print("모델 가중치 저장 완료!")
|
| 278 |
|
| 279 |
+
def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperature=0.8, min_len=20):
|
| 280 |
model_input = text_to_ids(f"<start> {prompt}")
|
| 281 |
model_input = model_input[:max_len]
|
| 282 |
generated = list(model_input)
|