File size: 2,823 Bytes
de42d2c
c76e549
de42d2c
 
 
 
c76e549
 
 
 
 
 
 
de42d2c
 
 
 
 
c76e549
de42d2c
c76e549
 
de42d2c
c76e549
de42d2c
c76e549
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
86
87
88
89
90
91
---
base_model: google/gemma-4-e2b-it
tags:
- text-generation-inference
- transformers
- gemma4
- peft
- lora
- cybersecurity
- docker
- container-security
- devsecops
- cybersecurity
license: apache-2.0
language:
- en
---

# Gemma 4 E2B — Docker & Container Security Expert

A QLoRA fine-tuned version of [Gemma 4 E2B Instruct](https://huggingface.co/google/gemma-4-e2b-it) specialized in **docker & container security**.
Specialized in **Docker and container security**: image hardening, rootless containers, seccomp/AppArmor profiles, runtime threat detection, and container escape techniques and mitigations.

Part of the [rezaduty cybersecurity model family](https://huggingface.co/rezaduty).

---

## Expertise

- Docker daemon security and socket exposure risks
- Image scanning, distroless images, and minimal base images
- Rootless containers, user namespaces, and capability dropping
- Runtime security with Falco, seccomp, and AppArmor
- Container escape techniques and kernel exploit mitigations
- Dockerfile best practices and supply-chain integrity

---

## Model Details

| Property | Value |
|---|---|
| **Base model** | google/gemma-4-e2b-it (2B parameters) |
| **Fine-tuning method** | QLoRA (rank 16, α 16) |
| **Domain** | Docker & Container Security |
| **License** | Apache 2.0 |

---

## Usage

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

base_model = "google/gemma-4-e2b-it"
adapter    = "rezaduty/gemma4-e2b-docker-container-security"

tokenizer = AutoTokenizer.from_pretrained(adapter)
model = AutoModelForCausalLM.from_pretrained(
    base_model, torch_dtype=torch.bfloat16, device_map="auto"
)
model = PeftModel.from_pretrained(model, adapter)

messages = [
    {"role": "system", "content": [{"type": "text", "text": "You are an expert in Docker and container security. You provide deep, production-level answers on container hardening, image security, runtime protection, and container escape prevention."}]},
    {"role": "user",   "content": [{"type": "text", "text": "Your question here"}]},
]
inputs = tokenizer.apply_chat_template(
    messages, tokenize=True, add_generation_prompt=True, return_tensors="pt"
).to(model.device)
output = model.generate(inputs, max_new_tokens=512, temperature=0.7, top_p=0.9)
print(tokenizer.decode(output[0][inputs.shape[-1]:], skip_special_tokens=True))
```

---

## System Prompt

```
You are an expert in Docker and container security. You provide deep, production-level answers on container hardening, image security, runtime protection, and container escape prevention.
```

---

## See Also

- [General cybersecurity model](https://huggingface.co/rezaduty/gemma4-e2b-cybersecurity-interview) — full 646-example dataset
- [All rezaduty models](https://huggingface.co/rezaduty)