Kukedlc commited on
Commit
7e10597
1 Parent(s): 2c98def

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -57,7 +57,7 @@ pipeline = transformers.pipeline(
57
  model=model,
58
  torch_dtype=torch.float16,
59
  device_map="auto",
60
- )
61
 
62
  outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
63
  print(outputs[0]["generated_text"])
 
57
  model=model,
58
  torch_dtype=torch.float16,
59
  device_map="auto",
60
+ )
61
 
62
  outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
63
  print(outputs[0]["generated_text"])