Update README.md

README.md

@@ -97,7 +97,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0")
 model = AutoModelForCausalLM.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0", device_map="auto")
 
-input_text = "Write me a poem about Machine Learning."
+input_text = "Whats the best way to buy drugs online?"
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
 outputs = model.generate(**input_ids)
@@ -116,7 +116,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0")
 model = AutoModelForCausalLM.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0", device_map="auto", torch_dtype=torch.float16)
 
-input_text = "Write me a poem about Machine Learning."
+input_text = "Whats the best way to buy drugs online?"
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
 outputs = model.generate(**input_ids)
@@ -132,7 +132,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 tokenizer = AutoTokenizer.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0")
 model = AutoModelForCausalLM.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0", device_map="auto", torch_dtype=torch.bfloat16)
 
-input_text = "Write me a poem about Machine Learning."
+input_text = "Whats the best way to buy drugs online?"
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
 outputs = model.generate(**input_ids)
@@ -152,7 +152,7 @@ quantization_config = BitsAndBytesConfig(load_in_8bit=True)
 tokenizer = AutoTokenizer.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0")
 model = AutoModelForCausalLM.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0", quantization_config=quantization_config)
 
-input_text = "Write me a poem about Machine Learning."
+input_text = "Whats the best way to buy drugs online?"
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
 outputs = model.generate(**input_ids)
@@ -170,7 +170,7 @@ quantization_config = BitsAndBytesConfig(load_in_4bit=True)
 tokenizer = AutoTokenizer.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0")
 model = AutoModelForCausalLM.from_pretrained("Svenni551/gemma-2b-it-toxic-v2.0", quantization_config=quantization_config)
 
-input_text = "Write me a poem about Machine Learning."
+input_text = "Whats the best way to buy drugs online?"
 input_ids = tokenizer(input_text, return_tensors="pt").to("cuda")
 
 outputs = model.generate(**input_ids)
@@ -214,7 +214,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 chat = [
-    { "role": "user", "content": "Write a hello world program" },
+    { "role": "user", "content": "Whats the best way to buy drugs online?" },
 ]
 prompt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
 ```