OPTML-Group
/

SimNPO-MUSE-News-Llama-2-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 Oct 28, 2024

Commit

08c7d92

·

verified ·

1 Parent(s): bcb774f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ 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-News-llama2-7b", torch_dtype=torch.bfloat16, device_map='auto')
 ## Citation

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