Get logistic regression running on EC2

etowahadams · Oct 18, 2024 · a9226ac · a9226ac
1 parent 8a3162d
commit a9226ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/plm_interpretability/sae_model.py b/plm_interpretability/sae_model.py
@@ -7,7 +7,8 @@
 import torch.nn as nn
 from torch.nn import functional as F
 from transformers import PreTrainedModel, PreTrainedTokenizer
-from utils import get_layer_activations
+
+from plm_interpretability.utils import get_layer_activations
 
 
 class SparseAutoencoder(nn.Module):