Update README.md

README.md

@@ -60,15 +60,14 @@ You can run the smashed model with these steps:
 2. Load & run the model.
     ```python 
    from transformers import AutoModelForCausalLM, AutoTokenizer
-   from awq import AutoAWQForCausalLM
-
-   model = AutoAWQForCausalLM.from_quantized("PrunaAI/OpenLLM-France-Lucie-7B-Instruct-human-data-AWQ-4bit-smashed", trust_remote_code=True, device_map='auto')
-   tokenizer = AutoTokenizer.from_pretrained("OpenLLM-France/Lucie-7B-Instruct-human-data")
-    
-   input_ids = tokenizer("What is the color of prunes?,", return_tensors='pt').to(model.device)["input_ids"]
-    
-   outputs = model.generate(input_ids, max_new_tokens=216)
-   tokenizer.decode(outputs[0])
+  from awq import AutoAWQForCausalLM
+  
+  model = AutoAWQForCausalLM.from_quantized("PrunaAI/OpenLLM-France-Lucie-7B-AWQ-4bit-smashed", trust_remote_code=True, device_map='auto')
+  tokenizer = AutoTokenizer.from_pretrained("OpenLLM-France/Lucie-7B")
+  
+  input_ids = tokenizer("Quelle est la couleur des pruneaux?", return_tensors='pt').to(next(model.parameters()).device)["input_ids"]
+  outputs = model.generate(input_ids, max_new_tokens=216)
+  tokenizer.decode(outputs[0])
     ```
 
 ## Configurations