/usr/bin/python3 -m mlc_llm gen_config dist/models/stablelm-2-zephyr-1_6b --quantization q4f16_1 --conv-template stablelm-2 --output /tmp/tmpuyytyuw2 --context-window-size 4096 [2024-05-24 18:11:22] INFO auto_config.py:115: Found model configuration: dist/models/stablelm-2-zephyr-1_6b/config.json [2024-05-24 18:11:22] INFO auto_config.py:153: Found model type: stablelm. Use `--model-type` to override. [2024-05-24 18:11:22] INFO stablelm_model.py:49: context_window_size not found in config.json. Falling back to max_position_embeddings (4096) [2024-05-24 18:11:22] INFO stablelm_model.py:66: prefill_chunk_size defaults to 2048 [2024-05-24 18:11:22] INFO config.py:106: Overriding context_window_size from 4096 to 4096 [2024-05-24 18:11:22] INFO config.py:106: Overriding max_batch_size from 1 to 80 [2024-05-24 18:11:22] INFO gen_config.py:255: [generation_config.json] Setting bos_token_id: 100257 [2024-05-24 18:11:22] INFO gen_config.py:255: [generation_config.json] Setting eos_token_id: 100257 [2024-05-24 18:11:22] INFO gen_config.py:269: Not found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/tokenizer.model [2024-05-24 18:11:22] INFO gen_config.py:267: Found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/tokenizer.json. Copying to /tmp/tmpuyytyuw2/tokenizer.json [2024-05-24 18:11:22] INFO gen_config.py:267: Found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/vocab.json. Copying to /tmp/tmpuyytyuw2/vocab.json [2024-05-24 18:11:22] INFO gen_config.py:267: Found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/merges.txt. Copying to /tmp/tmpuyytyuw2/merges.txt [2024-05-24 18:11:22] INFO gen_config.py:269: Not found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/added_tokens.json [2024-05-24 18:11:22] INFO gen_config.py:267: Found tokenizer config: dist/models/stablelm-2-zephyr-1_6b/tokenizer_config.json. Copying to /tmp/tmpuyytyuw2/tokenizer_config.json [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting pad_token_id: 0 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting temperature: 0.7 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting presence_penalty: 0.0 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting frequency_penalty: 0.0 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting repetition_penalty: 1.0 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting top_p: 0.95 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting mean_gen_len: 128 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting max_gen_len: 512 [2024-05-24 18:11:23] INFO gen_config.py:80: [System default] Setting shift_fill_factor: 0.3 [2024-05-24 18:11:23] INFO gen_config.py:335: Dumping configuration file to: /tmp/tmpuyytyuw2/mlc-chat-config.json /usr/bin/python3 -m mlc_llm convert_weight dist/models/stablelm-2-zephyr-1_6b --quantization q4f16_1 --source-format auto --output /tmp/tmpuyytyuw2 [2024-05-24 18:11:23] INFO auto_config.py:115: Found model configuration: dist/models/stablelm-2-zephyr-1_6b/config.json [2024-05-24 18:11:26] INFO auto_device.py:79: Found device: cuda:0 [2024-05-24 18:11:27] INFO auto_device.py:88: Not found device: rocm:0 [2024-05-24 18:11:28] INFO auto_device.py:88: Not found device: metal:0 [2024-05-24 18:11:29] INFO auto_device.py:88: Not found device: vulkan:0 [2024-05-24 18:11:30] INFO auto_device.py:88: Not found device: opencl:0 [2024-05-24 18:11:30] INFO auto_device.py:35: Using device: cuda:0 [2024-05-24 18:11:30] INFO auto_weight.py:70: Finding weights in: dist/models/stablelm-2-zephyr-1_6b [2024-05-24 18:11:30] INFO auto_weight.py:136: Not found Huggingface PyTorch [2024-05-24 18:11:30] INFO auto_weight.py:143: Found source weight format: huggingface-safetensor. Source configuration: dist/models/stablelm-2-zephyr-1_6b/model.safetensors.index.json [2024-05-24 18:11:30] INFO auto_weight.py:106: Using source weight configuration: dist/models/stablelm-2-zephyr-1_6b/model.safetensors.index.json. Use `--source` to override. [2024-05-24 18:11:30] INFO auto_weight.py:110: Using source weight format: huggingface-safetensor. Use `--source-format` to override. [2024-05-24 18:11:30] INFO auto_config.py:153: Found model type: stablelm. Use `--model-type` to override. [2024-05-24 18:11:30] INFO stablelm_model.py:49: context_window_size not found in config.json. Falling back to max_position_embeddings (4096) [2024-05-24 18:11:30] INFO stablelm_model.py:66: prefill_chunk_size defaults to 2048 Weight conversion with arguments: --config dist/models/stablelm-2-zephyr-1_6b/config.json --quantization GroupQuantize(name='q4f16_1', kind='group-quant', group_size=32, quantize_dtype='int4', storage_dtype='uint32', model_dtype='float16', linear_weight_layout='NK', quantize_embedding=True, quantize_final_fc=True, num_elem_per_storage=8, num_storage_per_group=4, max_int_value=7) --model-type stablelm --device cuda:0 --source dist/models/stablelm-2-zephyr-1_6b/model.safetensors.index.json --source-format huggingface-safetensor --output /tmp/tmpuyytyuw2 Start storing to cache /tmp/tmpuyytyuw2 0%| | 0/220 [00:00