Update modeling_step_vl.py
Browse files- modeling_step_vl.py +0 -8
modeling_step_vl.py
CHANGED
|
@@ -558,11 +558,3 @@ class Step3VL10BForCausalLM(StepRoboticsPreTrainedModel, GenerationMixin):
|
|
| 558 |
return key[len("language_model."):], True
|
| 559 |
|
| 560 |
return key, False
|
| 561 |
-
|
| 562 |
-
|
| 563 |
-
# Register config/model so Auto classes can instantiate this implementation.
|
| 564 |
-
from transformers.models.auto.configuration_auto import AutoConfig
|
| 565 |
-
from transformers.models.auto.modeling_auto import AutoModelForCausalLM
|
| 566 |
-
|
| 567 |
-
AutoConfig.register("step_robotics", StepRoboticsConfig)
|
| 568 |
-
AutoModelForCausalLM.register(StepRoboticsConfig, Step3VL10BForCausalLM)
|
|
|
|
| 558 |
return key[len("language_model."):], True
|
| 559 |
|
| 560 |
return key, False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|