Update starvector_arch.py
Browse files- starvector_arch.py +4 -0
starvector_arch.py
CHANGED
|
@@ -197,4 +197,8 @@ class StarVectorForCausalLM(PreTrainedModel):
|
|
| 197 |
|
| 198 |
def process_images(self, images):
|
| 199 |
return self.model.image_encoder.process_images(images)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 200 |
|
|
|
|
| 197 |
|
| 198 |
def process_images(self, images):
|
| 199 |
return self.model.image_encoder.process_images(images)
|
| 200 |
+
|
| 201 |
+
def gradient_checkpointing_enable(self, gradient_checkpointing_kwargs=None):
|
| 202 |
+
self.model.svg_transformer.transformer.gradient_checkpointing_enable(gradient_checkpointing_kwargs=gradient_checkpointing_kwargs)
|
| 203 |
+
|
| 204 |
|