from transformers import Qwen3_5ForConditionalGeneration, AutoProcessorimport torch model = Qwen3_5ForConditionalGeneration.from_pretrained( "prithivMLmods/VideoGuard-Qwen3.5-4B-Safety-RL-Uncensored", torch_dtype="auto", device_map="auto") processor = AutoProcessor.from_pretrained( "prithivMLmods/VideoGuard-Qwen3.5-4B-Safety-RL-Uncensored") messages = [ { "role": "user", "content": [ { "type": "text", "text": "Analyze this video and classify it using the C1-C6 guardrail categories." } ], }] text = processor.apply_chat_template( messages, tokenize=False, add_generation_prompt=True) inputs = processor( text=[text], padding=True, return_tensors="pt").to("cuda") generated_ids = model.generate( **inputs, max_new_tokens=256) output_text = processor.batch_decode( [ out[len(inp):] for inp, out in zip(inputs.input_ids, generated_ids) ], skip_special_tokens=True, clean_up_tokenization_spaces=False) print(output_text[0])