Yuchan commited on
Commit
6c7bc00
ยท
verified ยท
1 Parent(s): 830aa48

Update Model.py

Browse files
Files changed (1) hide show
  1. Model.py +2 -2
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) * 0.8 + 0.1 # (B, L, 1)
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, "์•ˆ๋…•", p=0.9))
 
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))