/home/hangyu5/anaconda3/envs/llama_factory/lib/python3.11/site-packages/trl/trainer/ppo_config.py:141: UserWarning: The `optimize_cuda_cache` arguement will be deprecated soon, please use `optimize_device_cache` instead.
  warnings.warn(
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file vocab.json
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file merges.txt
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file tokenizer.json
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file added_tokens.json
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file special_tokens_map.json
[INFO|tokenization_utils_base.py:2024] 2023-12-19 12:57:02,821 >> loading file tokenizer_config.json
[WARNING|logging.py:314] 2023-12-19 12:57:02,875 >> Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
[INFO|configuration_utils.py:737] 2023-12-19 12:57:02,875 >> loading configuration file ./models/phi-2/config.json
[INFO|configuration_utils.py:737] 2023-12-19 12:57:02,877 >> loading configuration file ./models/phi-2/config.json
[INFO|configuration_utils.py:802] 2023-12-19 12:57:02,877 >> Model config PhiConfig {
  "_name_or_path": "./models/phi-2",
  "activation_function": "gelu_new",
  "architectures": [
    "PhiForCausalLM"
  ],
  "attn_pdrop": 0.0,
  "auto_map": {
    "AutoConfig": "configuration_phi.PhiConfig",
    "AutoModelForCausalLM": "modeling_phi.PhiForCausalLM"
  },
  "embd_pdrop": 0.0,
  "flash_attn": false,
  "flash_rotary": false,
  "fused_dense": false,
  "img_processor": null,
  "initializer_range": 0.02,
  "layer_norm_epsilon": 1e-05,
  "model_type": "phi-msft",
  "n_embd": 2560,
  "n_head": 32,
  "n_head_kv": null,
  "n_inner": null,
  "n_layer": 32,
  "n_positions": 2048,
  "resid_pdrop": 0.1,
  "rotary_dim": 32,
  "tie_word_embeddings": false,
  "torch_dtype": "float16",
  "transformers_version": "4.36.1",
  "vocab_size": 51200
}

[INFO|modeling_utils.py:3329] 2023-12-19 12:57:02,908 >> loading weights file ./models/phi-2/model.safetensors.index.json
[INFO|modeling_utils.py:1341] 2023-12-19 12:57:02,908 >> Instantiating PhiForCausalLM model under default dtype torch.float16.
[INFO|configuration_utils.py:826] 2023-12-19 12:57:02,909 >> Generate config GenerationConfig {}

[INFO|configuration_utils.py:826] 2023-12-19 12:57:02,909 >> Generate config GenerationConfig {}


Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]
Loading checkpoint shards:  50%|█████     | 1/2 [00:00<00:00,  5.79it/s]
Loading checkpoint shards: 100%|██████████| 2/2 [00:00<00:00,  6.04it/s]
Loading checkpoint shards: 100%|██████████| 2/2 [00:00<00:00,  6.00it/s]
[INFO|modeling_utils.py:4173] 2023-12-19 12:57:03,336 >> All model checkpoint weights were used when initializing PhiForCausalLM.

[INFO|modeling_utils.py:4181] 2023-12-19 12:57:03,336 >> All the weights of PhiForCausalLM were initialized from the model checkpoint at ./models/phi-2.
If your task is similar to the task the model of the checkpoint was trained on, you can already use PhiForCausalLM for predictions without further training.
[INFO|configuration_utils.py:779] 2023-12-19 12:57:03,338 >> loading configuration file ./models/phi-2/generation_config.json
[INFO|configuration_utils.py:826] 2023-12-19 12:57:03,339 >> Generate config GenerationConfig {}

12/19/2023 12:57:03 - INFO - llmtuner.model.adapter - Fine-tuning method: LoRA
12/19/2023 12:57:04 - INFO - llmtuner.model.adapter - Merged 1 adapter(s).
12/19/2023 12:57:04 - INFO - llmtuner.model.adapter - Loaded adapter(s): ./models/sft/phi-2-sft-alpaca_gpt4_en-ep1-lora
12/19/2023 12:57:04 - INFO - llmtuner.model.loader - trainable params: 0 || all params: 2779683840 || trainable%: 0.0000
12/19/2023 12:57:04 - INFO - llmtuner.model.loader - This IS expected that the trainable params is 0 if you are using model for inference only.
[INFO|configuration_utils.py:483] 2023-12-19 12:57:04,105 >> Configuration saved in ./models/export/phi-2-sft-alpaca_gpt4_en-ep1/config.json
[INFO|configuration_utils.py:594] 2023-12-19 12:57:04,105 >> Configuration saved in ./models/export/phi-2-sft-alpaca_gpt4_en-ep1/generation_config.json
[INFO|modeling_utils.py:2390] 2023-12-19 12:57:10,095 >> The model is bigger than the maximum size per checkpoint (5GB) and is going to be split in 2 checkpoint shards. You can find where each parameters has been saved in the index located at ./models/export/phi-2-sft-alpaca_gpt4_en-ep1/model.safetensors.index.json.
[INFO|tokenization_utils_base.py:2432] 2023-12-19 12:57:10,096 >> tokenizer config file saved in ./models/export/phi-2-sft-alpaca_gpt4_en-ep1/tokenizer_config.json
[INFO|tokenization_utils_base.py:2441] 2023-12-19 12:57:10,097 >> Special tokens file saved in ./models/export/phi-2-sft-alpaca_gpt4_en-ep1/special_tokens_map.json