project2you commited on
Commit
4e1c940
1 Parent(s): 0aa5172

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -10
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))