Views
No views yet
1from transformers import pipeline
2model_id = "yujiepan/gemma-3-tiny-random"
3pipe = pipeline(
4 "image-text-to-text", model=model_id, device="cuda",
5 trust_remote_code=True, max_new_tokens=3,
6)
7messages = [
8 {
9 "role": "system",
10 "content": [{"type": "text", "text": "You are a helpful assistant."}]
11 },
12 {
13 "role": "user",
14 "content": [
15 {"type": "image", "url": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/p-blog/candy.JPG"},
16 {"type": "text", "text": "What animal is on the candy?"}
17 ]
18 }
19]
20output = pipe(text=messages, max_new_tokens=5)
21print(output)1import torch
2
3from transformers import (
4 AutoConfig,
5 AutoModelForCausalLM,
6 AutoProcessor,
7 AutoTokenizer,
8 Gemma3ForConditionalGeneration,
9 GenerationConfig,
10 pipeline,
11 set_seed,
12)
13
14source_model_id = "google/gemma-3-27b-it"
15save_folder = "/tmp/yujiepan/gemma-3-tiny-random"
16
17processor = AutoProcessor.from_pretrained(
18 source_model_id, trust_remote_code=True,
19)
20processor.save_pretrained(save_folder)
21
22config = AutoConfig.from_pretrained(
23 source_model_id, trust_remote_code=True,
24)
25config.text_config.hidden_size = 32
26config.text_config.intermediate_size = 128
27config.text_config.head_dim = 32
28config.text_config.num_attention_heads = 1
29config.text_config.num_key_value_heads = 1
30config.text_config.num_hidden_layers = 2
31config.text_config.sliding_window_pattern = 2
32config.vision_config.hidden_size = 32
33config.vision_config.num_hidden_layers = 2
34config.vision_config.num_attention_heads = 1
35config.vision_config.intermediate_size = 128
36model = Gemma3ForConditionalGeneration(
37 config,
38).to(torch.bfloat16)
39for layer in model.language_model.model.layers:
40 print(layer.is_sliding)
41model.generation_config = GenerationConfig.from_pretrained(
42 source_model_id, trust_remote_code=True,
43)
44set_seed(42)
45with torch.no_grad():
46 for name, p in sorted(model.named_parameters()):
47 torch.nn.init.normal_(p, 0, 0.5)
48 print(name, p.shape)
49model.save_pretrained(save_folder)