Spaces:
Runtime error
Runtime error
project2you
commited on
Commit
•
4e1c940
1
Parent(s):
0aa5172
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +0,0 @@
|
|
1 |
-
from transformers import pipeline
|
2 |
-
|
3 |
-
pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, device='cuda:0')
|
4 |
-
|
5 |
-
with torch.autocast('cuda', dtype=torch.bfloat16):
|
6 |
-
print(
|
7 |
-
pipe('Here is a recipe for vegan banana bread:\n',
|
8 |
-
max_new_tokens=100,
|
9 |
-
do_sample=True,
|
10 |
-
use_cache=True))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|