Update modeling_virtual_cell.py

modeling_virtual_cell.py

@@ -1,18 +1,3 @@
-"""
-Virtual Cell Patient Model — HuggingFace release.
-
-Architecture: PaSCient (Cui et al., 2025). ConvergeBio contribution: training
-recipe, data scale, and model parameters.
-
-Usage:
-    from transformers import AutoModel
-    model = AutoModel.from_pretrained(
-        "ConvergeBio/virtual-cell-patient", trust_remote_code=True
-    )
-    # input_ids: [batch, num_cells, num_genes]  float32 log-normalized expression
-    out = model(input_ids=x)   # out.logits: [batch, num_classes]
-"""
-
 from typing import List, Optional
 
 import torch
@@ -94,8 +79,6 @@ class MLP(nn.Module):
 
 
 class MLPCellEmbedder(nn.Module):
-    # Thin wrapper that preserves the .encoder attribute name required
-    # for state-dict key compatibility with the checkpoint.
     def __init__(
         self,
         n_genes: int,