Update starvector_arch.py

starvector_arch.py

@@ -133,6 +133,7 @@ class StarVectorConfig(PretrainedConfig):
 class StarVectorForCausalLM(PreTrainedModel):
     config_class = StarVectorConfig
     _no_split_modules = []
+    _supports_flash_attn_2 = True
 
     def __init__(self, config: StarVectorConfig, **kwargs):
         super().__init__(config)