JuIm commited on
Commit
1f1792c
·
verified ·
1 Parent(s): a37c3f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained("JuIm/ProGemma") tokenizer = AutoTo
28
 
29
  progemma = pipeline("text-generation", model=model, tokenizer=tokenizer)
30
 
31
- sequence = progemma("<#bos#>", top_k=950, max_length=100, num_return_sequences=1, do_sample=True, repetition_penalty=1.2, eos_token_id=21, pad_token_id=22, bos_token_id=20)
32
 
33
  print(sequence)
34
 
 
28
 
29
  progemma = pipeline("text-generation", model=model, tokenizer=tokenizer)
30
 
31
+ sequence = progemma("\<bos>", top_k=950, max_length=100, num_return_sequences=1, do_sample=True, repetition_penalty=1.2, eos_token_id=21, pad_token_id=22, bos_token_id=20)
32
 
33
  print(sequence)
34