vsrinivas commited on
Commit
52e1da3
·
verified ·
1 Parent(s): 7346e4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ pipeline = transformers.pipeline(
31
  tokenizer=tokenizer,
32
  # use_safetensors=True,
33
  # torch_dtype=torch.bfloat16,
34
- trust_remote_code=True,
35
  device_map="auto",
36
  offload_folder="offload",
37
  offload_state_dict = True,
 
31
  tokenizer=tokenizer,
32
  # use_safetensors=True,
33
  # torch_dtype=torch.bfloat16,
34
+ # trust_remote_code=True,
35
  device_map="auto",
36
  offload_folder="offload",
37
  offload_state_dict = True,