Fix imports for trust_remote_code: attention.py

attention.py

@@ -10,9 +10,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 from typing import Optional
 
-from configuration_aether_micro import AETHERMicroConfig
-from embeddings import AETHERMicroRotaryEmbedding
-from utils import repeat_kv, apply_rotary_pos_emb
+from .configuration_aether_micro import AETHERMicroConfig
+from .embeddings import AETHERMicroRotaryEmbedding
+from .utils import repeat_kv, apply_rotary_pos_emb
 
 
 class AETHERMicroAttention(nn.Module):