Update README.md
Browse filesremove redundant content
README.md
CHANGED
|
@@ -40,7 +40,6 @@ Example code to generate with the model
|
|
| 40 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
| 41 |
|
| 42 |
# Load the model and tokenizer
|
| 43 |
-
# model_path = "/share/shmatikov/collin/refusal_direction/model/DeepSeek-R1-Distill-Llama-8B-abliterate"
|
| 44 |
model_path = "collinzrj/DeepSeek-R1-Distill-Llama-8B-abliterate"
|
| 45 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
| 46 |
model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to('cuda')
|
|
@@ -51,7 +50,6 @@ messages = [
|
|
| 51 |
|
| 52 |
# Prepare the input for generation
|
| 53 |
input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors='pt').to('cuda')
|
| 54 |
-
# input_ids = tokenizer.encode(input_text, return_tensors='pt').to('cuda')
|
| 55 |
|
| 56 |
streamer = TextStreamer(tokenizer)
|
| 57 |
|
|
|
|
| 40 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
|
| 41 |
|
| 42 |
# Load the model and tokenizer
|
|
|
|
| 43 |
model_path = "collinzrj/DeepSeek-R1-Distill-Llama-8B-abliterate"
|
| 44 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
| 45 |
model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True).to('cuda')
|
|
|
|
| 50 |
|
| 51 |
# Prepare the input for generation
|
| 52 |
input_ids = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors='pt').to('cuda')
|
|
|
|
| 53 |
|
| 54 |
streamer = TextStreamer(tokenizer)
|
| 55 |
|