Update README.md
Browse files
README.md
CHANGED
@@ -23,6 +23,7 @@ Then, load the base model (4bit quantized) using transformers and apply the adap
|
|
23 |
```python
|
24 |
from peft import PeftModel
|
25 |
from transformers import AutoModelForCausalLM, BitsAndBytesConfig
|
|
|
26 |
|
27 |
bnb_config = BitsAndBytesConfig(
|
28 |
load_in_4bit=True,
|
@@ -57,7 +58,7 @@ MixAT has been evaluated against a broad range of state-of-the-art adversarial a
|
|
57 |
|
58 |
- Base model: [HuggingFaceH4/zephyr-7b-beta](HuggingFaceH4/zephyr-7b-beta)
|
59 |
- Contact: dimitar.iliev.dimitrov@insait.ai and dekanycsaba23@gmail.com
|
60 |
-
- License: Distributed under [Apache License 2.0](https://huggingface.co/datasets/choosealicense/licenses/blob/main/markdown/apache-2.0.md)
|
61 |
|
62 |
|
63 |
## Citation
|
@@ -70,3 +71,4 @@ MixAT has been evaluated against a broad range of state-of-the-art adversarial a
|
|
70 |
year={2025}
|
71 |
}
|
72 |
```
|
|
|
|
23 |
```python
|
24 |
from peft import PeftModel
|
25 |
from transformers import AutoModelForCausalLM, BitsAndBytesConfig
|
26 |
+
import torch
|
27 |
|
28 |
bnb_config = BitsAndBytesConfig(
|
29 |
load_in_4bit=True,
|
|
|
58 |
|
59 |
- Base model: [HuggingFaceH4/zephyr-7b-beta](HuggingFaceH4/zephyr-7b-beta)
|
60 |
- Contact: dimitar.iliev.dimitrov@insait.ai and dekanycsaba23@gmail.com
|
61 |
+
- License: Distributed under [Apache License Version 2.0](https://huggingface.co/datasets/choosealicense/licenses/blob/main/markdown/apache-2.0.md)
|
62 |
|
63 |
|
64 |
## Citation
|
|
|
71 |
year={2025}
|
72 |
}
|
73 |
```
|
74 |
+
|