Skip to main content

vision_inference

Method
This is a method of the TrainVisionModel class in the train_vision module.

Signature

def vision_inference(instruction: Any, image: Any) -> Any

Parameters

instruction
Any
required
No description available.
image
Any
required
No description available.

Returns

Returns
Any
The result of the operation.

Uses

  • FastVisionModel.for_inference
  • apply_chat_template
  • to
  • hf_tokenizer
  • generate
  • batch_decode

Used By

Source

View on GitHub

praisonai/train_vision.py at line 194