deepapaikar commited on
Commit
d220dac
·
verified ·
1 Parent(s): d7ed0bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -13,7 +13,6 @@ pipeline = transformers.pipeline(
13
  "text-generation",
14
  model=model,
15
  torch_dtype=torch.float16,
16
- device_map="auto",
17
  )
18
 
19
  tokenizer = AutoTokenizer.from_pretrained(model)
 
13
  "text-generation",
14
  model=model,
15
  torch_dtype=torch.float16,
 
16
  )
17
 
18
  tokenizer = AutoTokenizer.from_pretrained(model)