Bug in control.py (#18)

If `model` has an attribute called `model`, then you should get the layers via `model.model.layers` not `model.layers`

repeng/control.py

@@ -204,7 +204,7 @@ def model_layer_list(model: ControlModel | PreTrainedModel) -> torch.nn.ModuleLi
         model = model.model
 
     if hasattr(model, "model"):  # mistral-like
-        return model.layers
+        return model.model.layers
     elif hasattr(model, "transformer"):  # gpt-2-like
         return model.transformer.h
     else: