Skip to content

Commit 230ef8f

Browse files
committed
add processor
1 parent 86ba583 commit 230ef8f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1410,12 +1410,12 @@ model = SwarmFormerModel.from_pretrained("${model.id}")
14101410

14111411
export const univa = (model: ModelData): string[] => [
14121412
`from univa.models.qwen2p5vl.modeling_univa_qwen2p5vl import UnivaQwen2p5VLForConditionalGeneration
1413-
1414-
model = UnivaQwen2p5VLForConditionalGeneration.from_pretrained(
1415-
${model.id},
1413+
model = UnivaQwen2p5VLForConditionalGeneration.from_pretrained(
1414+
"${model.id}",
14161415
torch_dtype=torch.bfloat16,
14171416
attn_implementation="flash_attention_2",
14181417
).to("cuda")
1418+
processor = AutoProcessor.from_pretrained("${model.id}")
14191419
`,
14201420
];
14211421

0 commit comments

Comments
 (0)