OPTML-Group
/

SimNPO-MUSE-Books-iclm-7b

Text Generation

machine-unlearning

large-language-models

trustworthy-machine-learning

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

a-F1 commited on 20 days ago

Commit

dc2869c

•

1 Parent(s): 37bb664

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -23,7 +23,8 @@ This model uses the `SimNPO` unlearning algorithm with the following parameters:
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model = AutoModelForCausalLM.from_pretrained("OPTML-Group/SimNPO-MUSE-Books-Llama-2-7b", torch_dtype=torch.bfloat16, device_map='auto')
 ## Citation

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model = AutoModelForCausalLM.from_pretrained("OPTML-Group/SimNPO-MUSE-Books-iclm-7b", torch_dtype=torch.bfloat16, device_map='auto')
+```
 ## Citation