Add Wall-X serving and Turtle2 TCP WebSocket bridge
Pre-commit / pre-commit (push) Canceled after 0s
Pre-commit / pre-commit (push) Canceled after 0s
This commit is contained in:
@@ -94,6 +94,10 @@ class ModelAdapter(ABC):
|
||||
"""Load pretrained weights, resize embeddings, set normalizers, etc."""
|
||||
...
|
||||
|
||||
def finalize_model_after_weight_load(self, model, model_config):
|
||||
"""Apply wrappers that must be created after loading base model weights."""
|
||||
return model
|
||||
|
||||
# ---- FSDP wrapping ----
|
||||
@abstractmethod
|
||||
def get_transformer_layer_cls(self):
|
||||
|
||||
Reference in New Issue
Block a user