We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 86ba583 commit 230ef8fCopy full SHA for 230ef8f
packages/tasks/src/model-libraries-snippets.ts
@@ -1410,12 +1410,12 @@ model = SwarmFormerModel.from_pretrained("${model.id}")
1410
1411
export const univa = (model: ModelData): string[] => [
1412
`from univa.models.qwen2p5vl.modeling_univa_qwen2p5vl import UnivaQwen2p5VLForConditionalGeneration
1413
-
1414
-model = UnivaQwen2p5VLForConditionalGeneration.from_pretrained(
1415
- ${model.id},
+ model = UnivaQwen2p5VLForConditionalGeneration.from_pretrained(
+ "${model.id}",
1416
torch_dtype=torch.bfloat16,
1417
attn_implementation="flash_attention_2",
1418
).to("cuda")
+ processor = AutoProcessor.from_pretrained("${model.id}")
1419
`,
1420
];
1421
0 commit comments