Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
@@ -149,7 +149,7 @@ print("모델 가중치 로드 완료!")
|
|
149 |
|
150 |
|
151 |
def generate_text_sample(model, prompt, max_len=100, max_gen=98,
|
152 |
-
temperature=0.85, top_k=65, top_p=0.
|
153 |
model_input = text_to_ids(f"{prompt}")
|
154 |
model_input = model_input[:max_len]
|
155 |
generated = list(model_input)
|
|
|
149 |
|
150 |
|
151 |
def generate_text_sample(model, prompt, max_len=100, max_gen=98,
|
152 |
+
temperature=0.85, top_k=65, top_p=0.9, min_len=12):
|
153 |
model_input = text_to_ids(f"{prompt}")
|
154 |
model_input = model_input[:max_len]
|
155 |
generated = list(model_input)
|