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

Changed float type to float.32

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.float64,
12
  device_map="auto",
13
  )
14
 
 
8
  pipeline = transformers.pipeline(
9
  "text-generation",
10
  model=model,
11
+ torch_dtype=torch.float32,
12
  device_map="auto",
13
  )
14