Translation
PEFT
English
Dutch
bnjmnmarie commited on
Commit
f80f1a7
1 Parent(s): cb8a977

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -54,9 +54,9 @@ model = PeftModel.from_pretrained(model, "kaitchup/Llama-2-7b-mt-Dutch-to-Englis
54
  Then, run the model as follows:
55
 
56
  ```
57
- my_text_translate = "Het is voor jou."
58
 
59
- prompt = my_text_translate+"###> "
60
 
61
  tokenized_input = tokenizer(prompt, return_tensors="pt")
62
  input_ids = tokenized_input["input_ids"].cuda()
@@ -71,7 +71,7 @@ generation_output = model.generate(
71
  )
72
  for seq in generation_output.sequences:
73
  output = tokenizer.decode(seq, skip_special_tokens=True)
74
- print(output.split("###> ")[1].strip())
75
  ```
76
 
77
 
 
54
  Then, run the model as follows:
55
 
56
  ```
57
+ my_text = "Het is voor jou."
58
 
59
+ prompt = my_text+" ###>"
60
 
61
  tokenized_input = tokenizer(prompt, return_tensors="pt")
62
  input_ids = tokenized_input["input_ids"].cuda()
 
71
  )
72
  for seq in generation_output.sequences:
73
  output = tokenizer.decode(seq, skip_special_tokens=True)
74
+ print(output.split("###>")[1].strip())
75
  ```
76
 
77