Kush1 commited on
Commit
d1edbc7
·
1 Parent(s): 80b0c7e

changes in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -17,6 +17,7 @@ model = "meta-llama/Llama-2-13b-chat-hf"
17
  pipeline = transformers.pipeline(
18
  "text-generation",
19
  model=model,
 
20
  device_map="auto",
21
  do_sample=True,
22
  token=HF_TOKEN,
 
17
  pipeline = transformers.pipeline(
18
  "text-generation",
19
  model=model,
20
+ torch_dtype=torch.float32,
21
  device_map="auto",
22
  do_sample=True,
23
  token=HF_TOKEN,