acbdkk commited on
Commit
78cd5a2
1 Parent(s): 23aea07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ tokenizer = transformers.AutoTokenizer.from_pretrained(model)
7
  pipeline = transformers.pipeline(
8
  "text-generation",
9
  model=model,
10
- torch_dtype=torch.bfloat16,
11
  )
12
 
13
  prompt = "What is 6534+2343?"
 
7
  pipeline = transformers.pipeline(
8
  "text-generation",
9
  model=model,
10
+ torch_dtype=torch.float32,
11
  )
12
 
13
  prompt = "What is 6534+2343?"