KazukiIz commited on
Commit
7b1e8c8
·
verified ·
1 Parent(s): f2a82cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -7,14 +7,14 @@ import tempfile
7
  # モデルのロード
8
  model_name = "elyza/ELYZA-japanese-CodeLlama-7b-instruct"
9
 
10
- bnb_config = BitsAndBytesConfig(
11
- load_in_8bit=True, # 8ビット量子化を有効化
12
- )
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  model = AutoModelForCausalLM.from_pretrained(
16
  model_name,
17
- quantization_config=bnb_config,
18
  # offload_folder="./offload"
19
  device_map="auto" # 自動的にGPUまたはCPUに割り当て
20
  )
 
7
  # モデルのロード
8
  model_name = "elyza/ELYZA-japanese-CodeLlama-7b-instruct"
9
 
10
+ # bnb_config = BitsAndBytesConfig(
11
+ # load_in_8bit=True, # 8ビット量子化を有効化
12
+ #)
13
 
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
  model = AutoModelForCausalLM.from_pretrained(
16
  model_name,
17
+ # quantization_config=bnb_config,
18
  # offload_folder="./offload"
19
  device_map="auto" # 自動的にGPUまたはCPUに割り当て
20
  )