[LLM] IPEX auto importer set on by default (#9832)

* Set BIGDL_IMPORT_IPEX default to True

* Remove import intel_extension_for_pytorch as ipex from GPU example
This commit is contained in:
Ziteng Zhang 2024-01-04 13:33:29 +08:00 committed by GitHub
parent 4ceefc9b18
commit 05b681fa85
64 changed files with 1 additions and 64 deletions

View file

@ -33,7 +33,6 @@ world_size = get_int_from_env(["WORLD_SIZE","PMI_SIZE"], "1")
from bigdl.llm import optimize_model
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -17,7 +17,6 @@
import torch
import time
import argparse
import intel_extension_for_pytorch as ipex
from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import AutoTokenizer

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -17,7 +17,6 @@
import torch
import time
import argparse
import intel_extension_for_pytorch as ipex
from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import LlamaTokenizer, GPTQConfig

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -14,7 +14,6 @@
# limitations under the License.
#
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -22,7 +22,6 @@ from transformers import AutoTokenizer
from transformers.generation import GenerationConfig
from bigdl.llm.transformers import AutoModelForCausalLM
import intel_extension_for_pytorch as ipex
torch.manual_seed(1234)

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -17,7 +17,6 @@
import os
import torch
import time
import intel_extension_for_pytorch as ipex
import argparse
import numpy as np
import inquirer

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import argparse
from bigdl.llm.transformers import AutoModelForCausalLM
from transformers import AutoTokenizer, TextGenerationPipeline

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
import numpy as np

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -14,7 +14,6 @@
# limitations under the License.
#
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -23,7 +23,6 @@ import time
import warnings
from typing import Dict, List, Optional, Union
import intel_extension_for_pytorch as ipex
import torch
import torch.nn as nn
from accelerate import init_empty_weights

View file

@ -34,7 +34,6 @@
import argparse
import torch
import intel_extension_for_pytorch as ipex
import time
from transformers import AutoModelForCausalLM

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -22,7 +22,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig
from bigdl.llm import optimize_model
import intel_extension_for_pytorch as ipex
torch.manual_seed(1234)

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse

View file

@ -15,7 +15,6 @@
#
import torch
import intel_extension_for_pytorch as ipex
import time
import argparse
from bigdl.llm import optimize_model

View file

@ -46,7 +46,6 @@ from peft import (
)
from utils.prompter import Prompter
import intel_extension_for_pytorch as ipex
from transformers import BitsAndBytesConfig
from bigdl.llm.transformers import AutoModelForCausalLM
# import them from bigdl.llm.transformers.qlora to get a BigDL-LLM compatible Peft model

View file

@ -19,7 +19,6 @@ import os
import transformers
from transformers import LlamaTokenizer
import intel_extension_for_pytorch as ipex
from peft import LoraConfig
from transformers import BitsAndBytesConfig
from bigdl.llm.transformers.qlora import get_peft_model, prepare_model_for_kbit_training

View file

@ -24,7 +24,7 @@ from .optimize import optimize_model
import os
# Default is false, set to true to auto importing Intel Extension for PyTorch.
BIGDL_IMPORT_IPEX = os.getenv("BIGDL_IMPORT_IPEX", 'False').lower() in ('true', '1', 't')
BIGDL_IMPORT_IPEX = os.getenv("BIGDL_IMPORT_IPEX", 'True').lower() in ('true', '1', 't')
if BIGDL_IMPORT_IPEX:
# Import Intel Extension for PyTorch as ipex if XPU version is installed
from .utils.ipex_importer import ipex_importer