anirudh-sub commited on
Commit
e5a60a4
·
1 Parent(s): 7379b19

Changed the tensor float type to float.64

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model)
8
  pipeline = transformers.pipeline(
9
  "text-generation",
10
  model=model,
11
- torch_dtype=torch.float16,
12
  device_map="auto",
13
  )
14
 
 
8
  pipeline = transformers.pipeline(
9
  "text-generation",
10
  model=model,
11
+ torch_dtype=torch.float64,
12
  device_map="auto",
13
  )
14