dinhdat1110 commited on
Commit
7eff01a
1 Parent(s): 4e4c6d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def model_loading_pipeline():
28
  "text-generation",
29
  model=model_id,
30
  model_kwargs={
31
- "torch_dtype": torch.float16,
32
  },
33
  streamer=streamer,
34
  )
 
28
  "text-generation",
29
  model=model_id,
30
  model_kwargs={
31
+ "torch_dtype": torch.float32,
32
  },
33
  streamer=streamer,
34
  )