This commit is contained in:
parent
d57cc03b60
commit
3e237e9c82
|
|
@ -1,5 +1,8 @@
|
|||
python - <<'PY'
|
||||
import warnings, torch
|
||||
warnings.filterwarnings("ignore", category=UserWarning, message=".*TypedStorage is deprecated.*")
|
||||
warnings.filterwarnings("ignore", category=UserWarning, message="Was asked to gather.*")
|
||||
|
||||
from datasets import load_dataset
|
||||
from transformers import (
|
||||
AutoTokenizer, AutoModelForCausalLM,
|
||||
|
|
|
|||
Loading…
Reference in New Issue