Yuchan
commited on
Update Model.py
Browse files
Model.py
CHANGED
|
@@ -243,7 +243,7 @@ class LoSoU(layers.Layer):
|
|
| 243 |
score = g_q * g_k
|
| 244 |
|
| 245 |
# ๋์ alpha ๊ณ์ฐ: (B, L, d_model) -> (B, L, 1)
|
| 246 |
-
alpha_dynamic = self.alpha_linear(x_f32)
|
| 247 |
# ํ์์ alpha_dynamic์ ๋ํ ํ์ฒ๋ฆฌ (์: min/max ๋ฑ) ๊ฐ๋ฅ
|
| 248 |
# ex: alpha_dynamic = tf.clip_by_value(alpha_dynamic, 0.01, 0.99)
|
| 249 |
|
|
|
|
| 243 |
score = g_q * g_k
|
| 244 |
|
| 245 |
# ๋์ alpha ๊ณ์ฐ: (B, L, d_model) -> (B, L, 1)
|
| 246 |
+
alpha_dynamic = self.alpha_linear(x_f32) # (B, L, 1)
|
| 247 |
# ํ์์ alpha_dynamic์ ๋ํ ํ์ฒ๋ฆฌ (์: min/max ๋ฑ) ๊ฐ๋ฅ
|
| 248 |
# ex: alpha_dynamic = tf.clip_by_value(alpha_dynamic, 0.01, 0.99)
|
| 249 |
|