Update Model.py

Model.py

@@ -54,7 +54,7 @@ if not os.path.exists(DATA_PATH):
 
 if not os.path.exists(TOKENIZER_PATH):
     download_file(
-        "https://huggingface.co/Yuchan5386/Respiso/resolve/main/bpe.model?download=true",
+        "https://huggingface.co/Yuchan5386/inlam-100m/resolve/main/ko_unigram.model?download=true",
         TOKENIZER_PATH
     )
 
@@ -127,7 +127,7 @@ class SwiGLU(layers.Layer):
         out = self.W1(tf.nn.silu(a) * b)     
         return tf.cast(out, x.dtype)
 
-class SparseCausalAttention(Layer):
+class SparseCausalAttention(tf.keras.layers.Layer):
     def __init__(self, num_heads, head_dim, window_size=16, **kwargs):
         super().__init__(**kwargs)
         self.num_heads = num_heads