loubb
/

aria-medium-embedding

Model card Files Files and versions

Removed stray cuda call

#2

by justbruno - opened Dec 1, 2025

base: refs/heads/main

←

from: refs/pr/2

Discussion Files changed

Files changed (1) hide show

modeling_aria.py +1 -1

modeling_aria.py CHANGED Viewed

@@ -620,7 +620,7 @@ class AriaForSequenceEmbedding(AriaPreTrainedModel):
 def precompute_causal_mask(max_seq_len: int):
     return torch.tril(
         torch.ones(max_seq_len, max_seq_len, dtype=torch.bool)
-    ).cuda()
 def precompute_freqs_cis(

 def precompute_causal_mask(max_seq_len: int):
     return torch.tril(
         torch.ones(max_seq_len, max_seq_len, dtype=torch.bool)
+    )
 def precompute_freqs_cis(