Skip to content

Commit

Permalink
minor
Browse files Browse the repository at this point in the history
  • Loading branch information
pomonam committed Jun 26, 2024
1 parent d33eabe commit ab99c93
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion examples/openwebtext/generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,12 @@

# prompt = "Doctors are male and nurses are"
# prompt = "Machine learning can be defined as"
prompt = "Using a distributed database has many advantages"
# prompt = "Using a distributed database has many advantages."

prompt = """Story title: 3 llamas go for a walk
Summary: The 3 llamas crossed a bridge and something unexpected happened
Once upon a time"""

outputs = pipeline(prompt, max_length=64)
print("Prompt:")
Expand Down

0 comments on commit ab99c93

Please sign in to comment.