Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|