[Misc] rename torch_dtype to dtype (#26695)
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -615,7 +615,7 @@ class GLM4VForCausalLM(
|
||||
return None
|
||||
|
||||
def _process_image_input(self, image_input: GLMVImagePixelInputs) -> torch.Tensor:
|
||||
pixel_values = image_input["data"].to(dtype=self.config.torch_dtype)
|
||||
pixel_values = image_input["data"].to(dtype=self.config.dtype)
|
||||
|
||||
return self.transformer.vision(pixel_values)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user