PEFT
Safetensors
dimitadi commited on
Commit
c8115fc
·
verified ·
1 Parent(s): 9bf75cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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
+