***

layout: overview
slug: nemo-curator/nemo\_curator/models/qwen\_vl
title: nemo\_curator.models.qwen\_vl
------------------------------------

## Module Contents

### Classes

| Name                                                            | Description |
| --------------------------------------------------------------- | ----------- |
| [`LLM`](#nemo_curator-models-qwen_vl-LLM)                       | -           |
| [`QwenVL`](#nemo_curator-models-qwen_vl-QwenVL)                 | -           |
| [`SamplingParams`](#nemo_curator-models-qwen_vl-SamplingParams) | -           |

### Data

[`VLLM_AVAILABLE`](#nemo_curator-models-qwen_vl-VLLM_AVAILABLE)

[`_QWEN2_5_VL_MODEL_ID`](#nemo_curator-models-qwen_vl-_QWEN2_5_VL_MODEL_ID)

[`_QWEN2_5_VL_MODEL_REVISION`](#nemo_curator-models-qwen_vl-_QWEN2_5_VL_MODEL_REVISION)

[`_QWEN_VARIANTS_INFO`](#nemo_curator-models-qwen_vl-_QWEN_VARIANTS_INFO)

### API

<Anchor id="nemo_curator-models-qwen_vl-LLM">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    class nemo_curator.models.qwen_vl.LLM()
    ```
  </CodeBlock>
</Anchor>

<Indent />

<Anchor id="nemo_curator-models-qwen_vl-QwenVL">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    class nemo_curator.models.qwen_vl.QwenVL(
        model_dir: str,
        model_variant: str,
        caption_batch_size: int,
        fp8: bool = True,
        max_output_tokens: int = 512,
        model_does_preprocess: bool = False,
        disable_mmcache: bool = False,
        stage2_prompt_text: str | None = None,
        verbose: bool = False
    )
    ```
  </CodeBlock>
</Anchor>

<Indent>
  **Bases:** [ModelInterface](/nemo-curator/nemo_curator/models/base#nemo_curator-models-base-ModelInterface)

  <ParamField path="model_id_names" type="list[str]" />

  <ParamField path="pattern" type="= '(.*)(user_prompt)(.*)'" />

  <ParamField path="weight_file" />

  <Anchor id="nemo_curator-models-qwen_vl-QwenVL-download_weights_on_node">
    <CodeBlock showLineNumbers={false} wordWrap={true}>
      ```python
      nemo_curator.models.qwen_vl.QwenVL.download_weights_on_node(
          model_dir: str
      ) -> None
      ```
    </CodeBlock>
  </Anchor>

  <Indent>
    <Badge>
      classmethod
    </Badge>

    Download the weights for the QwenVL model on the node.
  </Indent>

  <Anchor id="nemo_curator-models-qwen_vl-QwenVL-generate">
    <CodeBlock showLineNumbers={false} wordWrap={true}>
      ```python
      nemo_curator.models.qwen_vl.QwenVL.generate(
          videos: list[dict[str, typing.Any]],
          generate_stage2_caption: bool = False,
          batch_size: int = 16
      ) -> list[str]
      ```
    </CodeBlock>
  </Anchor>

  <Indent />

  <Anchor id="nemo_curator-models-qwen_vl-QwenVL-setup">
    <CodeBlock showLineNumbers={false} wordWrap={true}>
      ```python
      nemo_curator.models.qwen_vl.QwenVL.setup() -> None
      ```
    </CodeBlock>
  </Anchor>

  <Indent />
</Indent>

<Anchor id="nemo_curator-models-qwen_vl-SamplingParams">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    class nemo_curator.models.qwen_vl.SamplingParams()
    ```
  </CodeBlock>
</Anchor>

<Indent />

<Anchor id="nemo_curator-models-qwen_vl-VLLM_AVAILABLE">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    nemo_curator.models.qwen_vl.VLLM_AVAILABLE = True
    ```
  </CodeBlock>
</Anchor>

<Anchor id="nemo_curator-models-qwen_vl-_QWEN2_5_VL_MODEL_ID">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    nemo_curator.models.qwen_vl._QWEN2_5_VL_MODEL_ID = 'Qwen/Qwen2.5-VL-7B-Instruct'
    ```
  </CodeBlock>
</Anchor>

<Anchor id="nemo_curator-models-qwen_vl-_QWEN2_5_VL_MODEL_REVISION">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    nemo_curator.models.qwen_vl._QWEN2_5_VL_MODEL_REVISION = 'cc59489'
    ```
  </CodeBlock>
</Anchor>

<Anchor id="nemo_curator-models-qwen_vl-_QWEN_VARIANTS_INFO">
  <CodeBlock showLineNumbers={false} wordWrap={true}>
    ```python
    nemo_curator.models.qwen_vl._QWEN_VARIANTS_INFO = {'qwen': _QWEN2_5_VL_MODEL_ID}
    ```
  </CodeBlock>
</Anchor>