Update modeling_llava_qwen2.py

modeling_llava_qwen2.py

@@ -11,6 +11,7 @@ from functools import partial, reduce
 from PIL import Image
 import torch.utils.checkpoint
 from torch import nn
+import torch
 from transformers.image_processing_utils import BatchFeature, get_size_dict
 from transformers.image_transforms import (convert_to_rgb, normalize, rescale, resize, to_channel_dimension_format, )
 from transformers.image_utils import (ChannelDimension, PILImageResampling, to_numpy_array, )
@@ -18,6 +19,8 @@ from transformers.modeling_outputs import BaseModelOutput, BaseModelOutputWithPo
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import ModelOutput
 
+torch.set_default_device('cuda')
+
 
 class SigLipImageProcessor:
     def __init__(self,