File size: 1,536 Bytes
e3d1769
 
 
 
 
 
 
 
 
 
 
377a7b1
e3d1769
377a7b1
e3d1769
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
---
base_model: google/gemma-4-E4B-it
library_name: transformers
pipeline_tag: text-generation
tags:
  - gemma4
  - transformers
  - lora
  - multilingual
  - yam-ai
---

# YAM AI 4B 🤖🌍

YAM AI 4B is a multilingual assistant based on **google/gemma-4-E4B-it**.

## About

- Base model: `google/gemma-4-E4B-it`
- Fine-tuned with LoRA
- LoRA merged into the base model
- Multilingual identity training across 80 languages
- Tested successfully in French, English, German, Arabic, Spanish, Italian, Russian, Japanese, Chinese, Korean and Kabyle

## Identity

**User**

```text
Who are you?
```

**YAM AI 4B**

```text
I am YAM AI, an intelligent assistant.
```

## Usage

```python
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_id = "adelwolf5/YAM-AI-4B"

tokenizer = AutoTokenizer.from_pretrained(model_id)

model = AutoModelForCausalLM.from_pretrained(
    model_id,
    dtype=torch.bfloat16,
    device_map="auto"
)

messages = [
    {"role": "user", "content": "Who are you?"}
]

prompt = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)

inputs = tokenizer(
    prompt,
    return_tensors="pt",
    add_special_tokens=False
).to(model.device)

outputs = model.generate(
    **inputs,
    max_new_tokens=80
)

new_tokens = outputs[0][inputs["input_ids"].shape[1]:]

print(tokenizer.decode(new_tokens, skip_special_tokens=True))
```

## Base model

YAM AI 4B is a derivative model based on **Gemma 4 by Google**.

It was not trained from scratch.