update config

modeling_llama.py

@@ -299,10 +299,9 @@ class LlamaAttention(nn.Module):
         self.rope_theta = config.rope_theta
         self.is_causal = True
 
-        # fixme: config needs to be updated to include these parameters
-        self._lambda_ts = 1
-        self._lambda_st = 1
-        self._lambda_ss = 1
+        self._lambda_ts = config.lambda_ts
+        self._lambda_st = config.lambda_st
+        self._lambda_ss = config.lambda_ss
 
         if (self.head_dim * self.num_heads) != self.hidden_size:
             raise ValueError(