plannist
commited on
Commit
·
0abd453
1
Parent(s):
1be9318
device delete
Browse files
app.py
CHANGED
|
@@ -15,7 +15,6 @@ pipe = pipeline(
|
|
| 15 |
"text-generation",
|
| 16 |
model=model,
|
| 17 |
tokenizer=tokenizer,
|
| 18 |
-
device=0,
|
| 19 |
max_new_tokens=256,
|
| 20 |
do_sample=True,
|
| 21 |
temperature=0.7,
|
|
|
|
| 15 |
"text-generation",
|
| 16 |
model=model,
|
| 17 |
tokenizer=tokenizer,
|
|
|
|
| 18 |
max_new_tokens=256,
|
| 19 |
do_sample=True,
|
| 20 |
temperature=0.7,
|