Yuchan
commited on
Update Model.py
Browse files
Model.py
CHANGED
|
@@ -366,7 +366,7 @@ history = model.fit(
|
|
| 366 |
model.save_weights("Cobra.weights.h5")
|
| 367 |
print("모델 가중치 저장 완료!")
|
| 368 |
|
| 369 |
-
def generate_text_topp(model, prompt, max_len=
|
| 370 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
| 371 |
model_input = model_input[:max_len]
|
| 372 |
generated = list(model_input)
|
|
|
|
| 366 |
model.save_weights("Cobra.weights.h5")
|
| 367 |
print("모델 가중치 저장 완료!")
|
| 368 |
|
| 369 |
+
def generate_text_topp(model, prompt, max_len=512, max_gen=512, p=0.9, temperature=0.8, min_len=20):
|
| 370 |
model_input = text_to_ids(f"<start> {prompt} <sep>")
|
| 371 |
model_input = model_input[:max_len]
|
| 372 |
generated = list(model_input)
|