Yuchan
commited on
Update Model.py
Browse files
Model.py
CHANGED
|
@@ -214,7 +214,7 @@ class LoSoU(layers.Layer):
|
|
| 214 |
score = g_q * g_k
|
| 215 |
|
| 216 |
# ๋์ alpha ๊ณ์ฐ: (B, L, d_model) -> (B, L, 1)
|
| 217 |
-
alpha_dynamic = self.alpha_linear(x_f32)
|
| 218 |
# ํ์์ alpha_dynamic์ ๋ํ ํ์ฒ๋ฆฌ (์: min/max ๋ฑ) ๊ฐ๋ฅ
|
| 219 |
# ex: alpha_dynamic = tf.clip_by_value(alpha_dynamic, 0.01, 0.99)
|
| 220 |
|
|
@@ -363,4 +363,4 @@ def generate_text_topp(model, prompt, max_len=100, max_gen=98, p=0.9, temperatur
|
|
| 363 |
return ids_to_text(generated)
|
| 364 |
|
| 365 |
print("\n\n===== ์์ฑ ๊ฒฐ๊ณผ =====")
|
| 366 |
-
print(generate_text_topp(model, "
|
|
|
|
| 214 |
score = g_q * g_k
|
| 215 |
|
| 216 |
# ๋์ alpha ๊ณ์ฐ: (B, L, d_model) -> (B, L, 1)
|
| 217 |
+
alpha_dynamic = self.alpha_linear(x_f32) # (B, L, 1)
|
| 218 |
# ํ์์ alpha_dynamic์ ๋ํ ํ์ฒ๋ฆฌ (์: min/max ๋ฑ) ๊ฐ๋ฅ
|
| 219 |
# ex: alpha_dynamic = tf.clip_by_value(alpha_dynamic, 0.01, 0.99)
|
| 220 |
|
|
|
|
| 363 |
return ids_to_text(generated)
|
| 364 |
|
| 365 |
print("\n\n===== ์์ฑ ๊ฒฐ๊ณผ =====")
|
| 366 |
+
print(generate_text_topp(model, "์ ๊ฐ ์ด๋ฐ๊ฐ ๋ฒ์ค๋ฅผ ํ์ผ ํด์ ์ค๋น ์ข ํด์ผ๊ฒ ์ด์. ์ฌ๋ฏธ์๋ ๋ํ์์ต๋๋ค!", p=0.9))
|