"
+ ],
+ "text/html": [
+ "\n",
+ " \n",
+ " \n",
+ "
\n",
+ " [60/60 01:41, Epoch 0/1]\n",
+ "
\n",
+ " \n",
+ " \n",
+ " \n",
+ " Step | \n",
+ " Training Loss | \n",
+ "
\n",
+ " \n",
+ " \n",
+ " \n",
+ " 1 | \n",
+ " 1.818600 | \n",
+ "
\n",
+ " \n",
+ " 2 | \n",
+ " 2.305800 | \n",
+ "
\n",
+ " \n",
+ " 3 | \n",
+ " 1.703400 | \n",
+ "
\n",
+ " \n",
+ " 4 | \n",
+ " 2.014700 | \n",
+ "
\n",
+ " \n",
+ " 5 | \n",
+ " 1.735100 | \n",
+ "
\n",
+ " \n",
+ " 6 | \n",
+ " 1.670600 | \n",
+ "
\n",
+ " \n",
+ " 7 | \n",
+ " 1.259200 | \n",
+ "
\n",
+ " \n",
+ " 8 | \n",
+ " 1.308900 | \n",
+ "
\n",
+ " \n",
+ " 9 | \n",
+ " 1.165100 | \n",
+ "
\n",
+ " \n",
+ " 10 | \n",
+ " 1.225300 | \n",
+ "
\n",
+ " \n",
+ " 11 | \n",
+ " 0.952900 | \n",
+ "
\n",
+ " \n",
+ " 12 | \n",
+ " 0.975100 | \n",
+ "
\n",
+ " \n",
+ " 13 | \n",
+ " 0.927200 | \n",
+ "
\n",
+ " \n",
+ " 14 | \n",
+ " 1.053400 | \n",
+ "
\n",
+ " \n",
+ " 15 | \n",
+ " 0.893100 | \n",
+ "
\n",
+ " \n",
+ " 16 | \n",
+ " 0.907500 | \n",
+ "
\n",
+ " \n",
+ " 17 | \n",
+ " 1.010400 | \n",
+ "
\n",
+ " \n",
+ " 18 | \n",
+ " 1.262300 | \n",
+ "
\n",
+ " \n",
+ " 19 | \n",
+ " 1.024500 | \n",
+ "
\n",
+ " \n",
+ " 20 | \n",
+ " 0.884400 | \n",
+ "
\n",
+ " \n",
+ " 21 | \n",
+ " 0.945600 | \n",
+ "
\n",
+ " \n",
+ " 22 | \n",
+ " 1.019000 | \n",
+ "
\n",
+ " \n",
+ " 23 | \n",
+ " 0.899700 | \n",
+ "
\n",
+ " \n",
+ " 24 | \n",
+ " 0.994800 | \n",
+ "
\n",
+ " \n",
+ " 25 | \n",
+ " 1.078800 | \n",
+ "
\n",
+ " \n",
+ " 26 | \n",
+ " 1.021300 | \n",
+ "
\n",
+ " \n",
+ " 27 | \n",
+ " 1.048400 | \n",
+ "
\n",
+ " \n",
+ " 28 | \n",
+ " 0.883900 | \n",
+ "
\n",
+ " \n",
+ " 29 | \n",
+ " 0.849500 | \n",
+ "
\n",
+ " \n",
+ " 30 | \n",
+ " 0.894900 | \n",
+ "
\n",
+ " \n",
+ " 31 | \n",
+ " 0.857800 | \n",
+ "
\n",
+ " \n",
+ " 32 | \n",
+ " 0.866900 | \n",
+ "
\n",
+ " \n",
+ " 33 | \n",
+ " 0.987500 | \n",
+ "
\n",
+ " \n",
+ " 34 | \n",
+ " 0.860600 | \n",
+ "
\n",
+ " \n",
+ " 35 | \n",
+ " 0.959300 | \n",
+ "
\n",
+ " \n",
+ " 36 | \n",
+ " 0.860900 | \n",
+ "
\n",
+ " \n",
+ " 37 | \n",
+ " 0.882400 | \n",
+ "
\n",
+ " \n",
+ " 38 | \n",
+ " 0.756700 | \n",
+ "
\n",
+ " \n",
+ " 39 | \n",
+ " 1.090100 | \n",
+ "
\n",
+ " \n",
+ " 40 | \n",
+ " 1.174900 | \n",
+ "
\n",
+ " \n",
+ " 41 | \n",
+ " 0.893200 | \n",
+ "
\n",
+ " \n",
+ " 42 | \n",
+ " 0.981300 | \n",
+ "
\n",
+ " \n",
+ " 43 | \n",
+ " 0.954800 | \n",
+ "
\n",
+ " \n",
+ " 44 | \n",
+ " 0.911400 | \n",
+ "
\n",
+ " \n",
+ " 45 | \n",
+ " 0.918000 | \n",
+ "
\n",
+ " \n",
+ " 46 | \n",
+ " 0.973800 | \n",
+ "
\n",
+ " \n",
+ " 47 | \n",
+ " 0.872000 | \n",
+ "
\n",
+ " \n",
+ " 48 | \n",
+ " 1.198100 | \n",
+ "
\n",
+ " \n",
+ " 49 | \n",
+ " 0.909500 | \n",
+ "
\n",
+ " \n",
+ " 50 | \n",
+ " 1.032100 | \n",
+ "
\n",
+ " \n",
+ " 51 | \n",
+ " 1.017500 | \n",
+ "
\n",
+ " \n",
+ " 52 | \n",
+ " 0.909200 | \n",
+ "
\n",
+ " \n",
+ " 53 | \n",
+ " 0.976500 | \n",
+ "
\n",
+ " \n",
+ " 54 | \n",
+ " 1.154300 | \n",
+ "
\n",
+ " \n",
+ " 55 | \n",
+ " 0.781000 | \n",
+ "
\n",
+ " \n",
+ " 56 | \n",
+ " 1.012800 | \n",
+ "
\n",
+ " \n",
+ " 57 | \n",
+ " 0.885200 | \n",
+ "
\n",
+ " \n",
+ " 58 | \n",
+ " 0.828500 | \n",
+ "
\n",
+ " \n",
+ " 59 | \n",
+ " 0.853500 | \n",
+ "
\n",
+ " \n",
+ " 60 | \n",
+ " 0.898100 | \n",
+ "
\n",
+ " \n",
+ "
"
+ ]
+ },
+ "metadata": {}
+ }
+ ]
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "exp.end()"
+ ],
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "8II72I92Bm9s",
+ "outputId": "6518c199-dff2-4c4a-f821-aefdf595fa2c"
+ },
+ "execution_count": 13,
+ "outputs": [
+ {
+ "output_type": "stream",
+ "name": "stderr",
+ "text": [
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m ---------------------------------------------------------------------------------------\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Comet.ml Experiment Summary\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m ---------------------------------------------------------------------------------------\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Data:\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m display_summary_level : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m name : apparent_pagoda_1033\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m url : https://www.comet.com/examples/comet-example-unsloth/cb280e1a2ac942cbab2343349d80282f\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Metrics [count] (min, max):\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m epoch [61] : (0.0001545595054095827, 0.00927357032457496)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m grad_norm [60] : (0.2999851107597351, 2.063112258911133)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m learning_rate [60] : (0.0, 0.0002)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m loss [60] : (0.7567, 2.3058)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m total_flos : 5726714157219840.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/epoch [61] : (0.0001545595054095827, 0.00927357032457496)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/grad_norm [60] : (0.2999851107597351, 2.063112258911133)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/learning_rate [60] : (0.0, 0.0002)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/loss [60] : (0.7567, 2.3058)\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/total_flos : 5726714157219840.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_loss : 1.0665242771307628\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_runtime : 109.8838\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_samples_per_second : 4.368\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_steps_per_second : 0.546\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_loss : 1.0665242771307628\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_runtime : 109.8838\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_samples_per_second : 4.368\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_steps_per_second : 0.546\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Others:\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m hasNestedParams : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Parameters:\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|dispatch_batches : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|even_batches : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|gradient_accumulation_kwargs : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|non_blocking : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|split_batches : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|use_seedable_sampler : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adafactor : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_beta1 : 0.9\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_beta2 : 0.999\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_epsilon : 1e-08\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|auto_find_batch_size : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|batch_eval_metrics : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|bf16 : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|bf16_full_eval : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|chars_per_token : \n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|data_seed : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_drop_last : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_num_workers : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_persistent_workers : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_pin_memory : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_prefetch_factor : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_batch_size : 1000\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_kwargs : {}\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_num_proc : 2\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_text_field : text\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_backend : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_broadcast_buffers : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_bucket_cap_mb : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_find_unused_parameters : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_timeout : 1800\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|debug : []\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|deepspeed : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|disable_tqdm : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dispatch_batches : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_eval : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_predict : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_train : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_accumulation_steps : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_delay : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_do_concat_batches : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_on_start : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_packing : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_steps : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_strategy : no\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_use_gather_object : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|evaluation_strategy : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16 : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_backend : auto\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_full_eval : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_opt_level : O1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp : []\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|min_num_params : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla_fsdp_grad_ckpt : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla_fsdp_v2 : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_min_num_params : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_transformer_layer_cls_to_wrap : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|full_determinism : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_accumulation_steps : 4\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_checkpointing : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_checkpointing_kwargs : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|greater_is_better : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|group_by_length : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|half_precision_backend : auto\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_always_push : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_model_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_private_repo : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_strategy : every_save\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_token : \n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ignore_data_skip : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_inputs_for_metrics : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_num_input_tokens_seen : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_tokens_per_second : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|jit_mode_eval : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|label_names : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|label_smoothing_factor : 0.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|learning_rate : 0.0002\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|length_column_name : length\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|load_best_model_at_end : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|local_rank : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_level : passive\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_level_replica : warning\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_on_each_node : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_dir : outputs/runs/Aug30_23-55-50_eba2d1e0a1d9\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_first_step : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_nan_inf_filter : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_steps : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_strategy : steps\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|lr_scheduler_kwargs : {}\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|lr_scheduler_type : linear\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_grad_norm : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_seq_length : 2048\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_steps : 60\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|metric_for_best_model : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|model_init_kwargs : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|mp_parameters : \n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|neftune_noise_alpha : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|no_cuda : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|num_of_sequences : 1024\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|num_train_epochs : 3.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim : adamw_8bit\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim_args : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim_target_modules : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|output_dir : outputs\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|overwrite_output_dir : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|packing : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|past_index : -1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_device_eval_batch_size : 8\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_device_train_batch_size : 2\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_gpu_eval_batch_size : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_gpu_train_batch_size : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|prediction_loss_only : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_model_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_organization : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_token : \n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ray_scope : last\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|remove_unused_columns : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|report_to : ['comet_ml', 'tensorboard']\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|restore_callback_states_from_checkpoint : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|resume_from_checkpoint : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|run_name : outputs\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_on_each_node : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_only_model : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_safetensors : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_steps : 500\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_strategy : steps\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_total_limit : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|seed : 3407\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|skip_memory_metrics : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|split_batches : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tf32 : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile_backend : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile_mode : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_empty_cache_steps : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torchdynamo : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tpu_metrics_debug : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tpu_num_cores : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_cpu : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_ipex : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_legacy_prediction_loop : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_liger : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_mps_device : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|warmup_ratio : 0.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|warmup_steps : 5\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|weight_decay : 0.01\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|_name_or_path : unsloth/meta-llama-3.1-8b-bnb-4bit\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|add_cross_attention : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|architectures : ['LlamaForCausalLM']\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|attention_bias : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|attention_dropout : 0.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|bad_words_ids : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|begin_suppress_tokens : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|bos_token_id : 128000\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|chunk_size_feed_forward : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|cross_attention_hidden_size : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|decoder_start_token_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|diversity_penalty : 0.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|do_sample : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|early_stopping : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|encoder_no_repeat_ngram_size : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|eos_token_id : 128001\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|exponential_decay_length_penalty : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|finetuning_task : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|forced_bos_token_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|forced_eos_token_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|hidden_act : silu\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|hidden_size : 4096\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|id2label|0 : LABEL_0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|id2label|1 : LABEL_1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|initializer_range : 0.02\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|intermediate_size : 14336\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|is_decoder : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|is_encoder_decoder : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|label2id|LABEL_0 : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|label2id|LABEL_1 : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|length_penalty : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|max_length : 20\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|max_position_embeddings : 131072\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|min_length : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|mlp_bias : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|model_type : llama\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|no_repeat_ngram_size : 0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_attention_heads : 32\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_beam_groups : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_beams : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_hidden_layers : 32\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_key_value_heads : 8\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_return_sequences : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_attentions : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_hidden_states : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_scores : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pad_token_id : 128004\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|prefix : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pretraining_tp : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|problem_type : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pruned_heads : {}\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_compute_dtype : bfloat16\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_quant_type : nf4\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_use_double_quant : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_enable_fp32_cpu_offload : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_has_fp16_weight : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_skip_modules : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_threshold : 6.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|load_in_4bit : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|load_in_8bit : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|quant_method : bitsandbytes\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|remove_invalid_values : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|repetition_penalty : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|return_dict : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|return_dict_in_generate : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rms_norm_eps : 1e-05\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|factor : 8.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|high_freq_factor : 4.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|low_freq_factor : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|original_max_position_embeddings : 8192\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|rope_type : llama3\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_theta : 500000.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|sep_token_id : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|suppress_tokens : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|task_specific_params : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|temperature : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tf_legacy_loss : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tie_encoder_decoder : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tie_word_embeddings : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tokenizer_class : None\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|top_k : 50\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|top_p : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|torch_dtype : bfloat16\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|torchscript : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|transformers_version : 4.44.2\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|typical_p : 1.0\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|unsloth_version : 2024.8\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|use_bfloat16 : False\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|use_cache : True\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|vocab_size : 128256\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m peft_config|default : LoraConfig(peft_type=, auto_mapping=None, base_model_name_or_path='unsloth/meta-llama-3.1-8b-bnb-4bit', revision=None, task_type=, inference_mode=False, r=16, target_modules={'k_proj', 'gate_proj', 'v_proj', 'o_proj', 'q_proj', 'down_proj', 'up_proj'}, lora_alpha=16, lora_dropout=0, fan_in_fan_out=False, bias='none', use_rslora=False, modules_to_save=None, init_lora_weights=True, layers_to_transform=None, layers_pattern=None, rank_pattern={}, alpha_pattern={}, megatron_config=None, megatron_core='megatron.core', loftq_config={}, use_dora=False, layer_replication=None, runtime_config=LoraRuntimeConfig(ephemeral_gpu_offload=False))\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Uploads:\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m environment details : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m filename : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m installed packages : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m model graph : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m notebook : 2\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m os packages : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m source_code : 1\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m \n",
+ "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n"
+ ]
+ }
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "source": [
+ "## ⚙ Inference"
+ ],
+ "metadata": {
+ "id": "0IdjUa_0pzPQ"
+ }
+ },
+ {
+ "cell_type": "code",
+ "source": [
+ "# alpaca_prompt = Copied from above\n",
+ "FastLanguageModel.for_inference(model)\n",
+ "inputs = tokenizer(\n",
+ "[\n",
+ " alpaca_prompt.format(\n",
+ " \"Continue the fibonnaci sequence.\", # instruction\n",
+ " \"1, 1, 2, 3, 5, 8\", # input\n",
+ " \"\", # output - leave this blank for generation\n",
+ " )\n",
+ "], return_tensors = \"pt\").to(\"cuda\")\n",
+ "\n",
+ "outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)\n",
+ "tokenizer.batch_decode(outputs)"
+ ],
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "76su-yv9Aad0",
+ "outputId": "8b9c2ccd-4597-41fc-d9ef-e665272fae05"
+ },
+ "execution_count": 12,
+ "outputs": [
+ {
+ "output_type": "execute_result",
+ "data": {
+ "text/plain": [
+ "['<|begin_of_text|>Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nContinue the fibonnaci sequence.\\n\\n### Input:\\n1, 1, 2, 3, 5, 8\\n\\n### Response:\\n13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597, 2584, 4181, 6765, 10946, 17711, 28657, 46368, 75025']"
+ ]
+ },
+ "metadata": {},
+ "execution_count": 12
+ }
+ ]
+ }
+ ]
+}
\ No newline at end of file
From 15858bea34f953c783d5eef5490a2e759a0ace49 Mon Sep 17 00:00:00 2001
From: Abby Morgan <86856445+anmorgan24@users.noreply.github.com>
Date: Thu, 5 Sep 2024 15:38:19 -0400
Subject: [PATCH 2/3] update authentication methods
---
.../unsloth/notebooks/Comet_and_unsloth.ipynb | 8146 +++++++++--------
1 file changed, 4074 insertions(+), 4072 deletions(-)
diff --git a/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb b/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
index a563f93..3ef0557 100644
--- a/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
+++ b/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
@@ -1,4067 +1,620 @@
{
- "nbformat": 4,
- "nbformat_minor": 0,
- "metadata": {
- "colab": {
- "provenance": [],
- "toc_visible": true,
- "machine_shape": "hm",
- "gpuType": "A100"
+ "cells": [
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "6RF2EQaKDoxr"
+ },
+ "source": [
+ "\n",
+ " \n",
+ ""
+ ]
},
- "kernelspec": {
- "name": "python3",
- "display_name": "Python 3"
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "KDjO5WbeDtp0"
+ },
+ "source": [
+ "# Comet and unsloth\n",
+ "\n",
+ "[Comet](https://www.comet.com/site/?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) is an MLOps platform designed to help data scientists and teams build better models faster! Comet provides tooling to track, explain, manage, and monitor your models in a single place! It works with Jupyter notebooks and scripts and-- most importantly--it's 100% free to get started!\n",
+ "\n",
+ "[unsloth](https://github.com/unslothai/unsloth) dramatically improves the speed and efficiency of LLM fine-tuning for models including Llama, Phi-3, Gemma, Mistral, and more. For a full listed of 100+ supported unsloth models, [see here](https://huggingface.co/unsloth).\n",
+ "\n",
+ "Instrument your torchtune training runs with Comet to start managing experiments with efficiency, reproducibility, and collaboration in mind.\n",
+ "\n",
+ "Find more information about [our integration with torchtune here](https://www.comet.com/docs/v2/integrations/third-party-tools/unsloth?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) or [learn about our other integrations here](https://www.comet.com/docs/v2/integrations?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth)."
+ ]
},
- "language_info": {
- "name": "python"
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "f0fhpJZgsYII"
+ },
+ "source": [
+ "## ⚙ Install and import dependencies"
+ ]
},
- "accelerator": "GPU",
- "widgets": {
- "application/vnd.jupyter.widget-state+json": {
- "77caf8ce2875463f86cb7e5b0e02d10f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_986a13810ebf4e11ac2d636013efc948",
- "IPY_MODEL_0902e41471fb40f7a7fb807397d1e780",
- "IPY_MODEL_ec2615ff447e4a72a359143122a58887"
- ],
- "layout": "IPY_MODEL_bb0052241a574fdbb6ca8c942d99e1eb"
- }
- },
- "986a13810ebf4e11ac2d636013efc948": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_1aff4958133640838a260a881e84bd66",
- "placeholder": "",
- "style": "IPY_MODEL_b920a8e467d64b629189d1c76541cd17",
- "value": "model.safetensors: 100%"
- }
- },
- "0902e41471fb40f7a7fb807397d1e780": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "danger",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_0fe442a6a70948a4a9b703352148d780",
- "max": 5702746390,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_5bf1ce41bf6c4b389bac20669bd64e3b",
- "value": 5702745847
- }
- },
- "ec2615ff447e4a72a359143122a58887": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_7d06574680cb41b9bfac0c3e47916d6e",
- "placeholder": "",
- "style": "IPY_MODEL_b6cd239b54c345feb11143f2364fcfaf",
- "value": " 5.70G/5.70G [00:13<00:00, 698MB/s]"
- }
+ {
+ "cell_type": "code",
+ "execution_count": 1,
+ "metadata": {
+ "id": "5zTSjY9r4cGc"
+ },
+ "outputs": [],
+ "source": [
+ "%%capture\n",
+ "!pip install comet_ml \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"\n",
+ "\n",
+ "# check torch version for Xformers (2.3 -> 0.0.27)\n",
+ "from torch import __version__; from packaging.version import Version as V\n",
+ "xformers = \"xformers==0.0.27\" if V(__version__) < V(\"2.4.0\") else \"xformers\"\n",
+ "!pip install --no-deps {xformers} trl peft accelerate bitsandbytes triton"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 5,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
},
- "bb0052241a574fdbb6ca8c942d99e1eb": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ "id": "p8GTqJOW88fI",
+ "outputId": "1c7f76ee-a9ff-418d-8067-4cb357f938e5"
+ },
+ "outputs": [
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Valid Comet API Key saved in /root/.comet.config (set COMET_CONFIG to change where it is saved).\n",
+ "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n",
+ "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m As you are running in a Jupyter environment, you will need to call `experiment.end()` when finished to ensure all metrics and code are logged before exiting.\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Couldn't find a Git repository in '/content' nor in any parent directory. Set `COMET_GIT_DIRECTORY` if your Git Repository is elsewhere.\n",
+ "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Experiment is live on comet.com https://www.comet.com/examples/comet-example-unsloth/cb280e1a2ac942cbab2343349d80282f\n",
+ "\n"
+ ]
+ }
+ ],
+ "source": [
+ "import comet_ml\n",
+ "\n",
+ "comet_ml.login()\n",
+ "exp = comet_ml.Experiment(project_name=\"comet-example-unsloth\", workspace=\"examples\")"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "from huggingface_hub import notebook_login\n",
+ "\n",
+ "notebook_login()"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "5bfxeJqKwpKj"
+ },
+ "source": [
+ "## ⚙ Download model"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 7,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
},
- "1aff4958133640838a260a881e84bd66": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ "id": "bZei5Dvw7_kM",
+ "outputId": "62f7a016-6c2b-4f76-ebf2-f2051b907932"
+ },
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.\n"
+ ]
+ }
+ ],
+ "source": [
+ "from unsloth import FastLanguageModel\n",
+ "import torch\n",
+ "\n",
+ "max_seq_length = 2048\n",
+ "dtype = None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+\n",
+ "load_in_4bit = True"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "NG--Sbr1n5Cv"
+ },
+ "source": [
+ "Find the full list of [100+ supported unsloth models here](https://huggingface.co/unsloth). For a full list of supported 4-bit models see [here](https://huggingface.co/collections/unsloth/load-4bit-models-4x-faster-659042e3a41c3cbad582e734)"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 8,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/",
+ "height": 284,
+ "referenced_widgets": [
+ "77caf8ce2875463f86cb7e5b0e02d10f",
+ "986a13810ebf4e11ac2d636013efc948",
+ "0902e41471fb40f7a7fb807397d1e780",
+ "ec2615ff447e4a72a359143122a58887",
+ "bb0052241a574fdbb6ca8c942d99e1eb",
+ "1aff4958133640838a260a881e84bd66",
+ "b920a8e467d64b629189d1c76541cd17",
+ "0fe442a6a70948a4a9b703352148d780",
+ "5bf1ce41bf6c4b389bac20669bd64e3b",
+ "7d06574680cb41b9bfac0c3e47916d6e",
+ "b6cd239b54c345feb11143f2364fcfaf",
+ "e38214ae388e4fdca976e2adeb30d20c",
+ "85d9ede0e7064521bfe7a92d54df211f",
+ "740857a56a134b2dba327dc8bdbc974a",
+ "725534abda3c462188fa27a978bfb59d",
+ "c61150dc57cb4279b38064e1bd82b3bc",
+ "6472e21298f44e6e9c5a3eefcedc59fc",
+ "adb028ffb1c045f9ade7be249a33b375",
+ "a1a32114bd2d4cb8882afebf56d64d41",
+ "8c6ee78067c0490292ee7845dd886a1b",
+ "6d28c2bb843a48908827b362ad094183",
+ "41d5a34977dd4488ba5b88c308301fe6",
+ "d0ce911bc4c845e88168f32781cf7923",
+ "718199222cfc412e9e1ee351b45c9db6",
+ "dcde9dfd56de417fb2057b3ccc9725f1",
+ "c749164ce965417db5411d00f9401623",
+ "61a290abfcb14c4f8b6cd9e453600bc5",
+ "7cce70a2af1a4fc2a04658264c37b0c7",
+ "096fb8d33b804a02b12bc085fa3c372f",
+ "c6345c375f474aaf88b7b73e662b20f2",
+ "b68053e4411747b0b12ab63d309aa765",
+ "76664f2e75744dbf875de8fd4fdbe2d4",
+ "05891b9099734df397dc8bb7396ef3e2",
+ "01f7a6c4eec04fc9b063add3719c0777",
+ "e4f2492f92a74794aec11bf7e2e8d554",
+ "cba30ecbcad14542b3acc82e4396275c",
+ "249258f9a1764d40a440a5846227059b",
+ "53c336629bca4e6cacfe69ffa316890a",
+ "b7c0946c9f1b40d5b469e28149808565",
+ "1bc726cc52084737a0269d01f9df6a16",
+ "51b3596a668742acab22e5a6659d2c55",
+ "30d75e70fb6e4c4db53137804d55295d",
+ "2c6dd56cf2f740cca814d5c6fa504d93",
+ "55e22c9351de46c4a192008dde020393",
+ "c333e3f23b8f4df6a7fd9f7a000a54ab",
+ "903f318a80d849098b11451a8c855d50",
+ "6b7e50badd5b4c7b979d20d7ffd1b132",
+ "d83d854a33ec4adabc9bef9c2b699a0b",
+ "1be75bb752214bdc8a3ef6e6976f3312",
+ "d03ff62190934187be5be033a6928545",
+ "1ec6fa8d08e642059f1d673f6260169f",
+ "23351e3d7e8741a1adad4c1292dab841",
+ "33c4f5595d924cd3a6d8c1575815a5c8",
+ "6c4cf50fd9ee484396e8272591d21ee9",
+ "124e70720a374644a6d360926059911f"
+ ]
},
- "b920a8e467d64b629189d1c76541cd17": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
+ "id": "L09m09Xs8Y2e",
+ "outputId": "4c67ce27-09cd-44dd-e987-7168a29ee335"
+ },
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "==((====))== Unsloth 2024.8: Fast Llama patching. Transformers = 4.44.2.\n",
+ " \\\\ /| GPU: NVIDIA A100-SXM4-40GB. Max memory: 39.564 GB. Platform = Linux.\n",
+ "O^O/ \\_/ \\ Pytorch: 2.4.0+cu121. CUDA = 8.0. CUDA Toolkit = 12.1.\n",
+ "\\ / Bfloat16 = TRUE. FA [Xformers = 0.0.27.post2. FA2 = False]\n",
+ " \"-____-\" Free Apache license: http://github.com/unslothai/unsloth\n",
+ "Unsloth: Fast downloading is enabled - ignore downloading bars which are red colored!\n"
+ ]
},
- "0fe442a6a70948a4a9b703352148d780": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "77caf8ce2875463f86cb7e5b0e02d10f",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "model.safetensors: 0%| | 0.00/5.70G [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "5bf1ce41bf6c4b389bac20669bd64e3b": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "e38214ae388e4fdca976e2adeb30d20c",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "generation_config.json: 0%| | 0.00/230 [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "7d06574680cb41b9bfac0c3e47916d6e": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "d0ce911bc4c845e88168f32781cf7923",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "tokenizer_config.json: 0%| | 0.00/50.6k [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "b6cd239b54c345feb11143f2364fcfaf": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "01f7a6c4eec04fc9b063add3719c0777",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "tokenizer.json: 0%| | 0.00/9.09M [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "e38214ae388e4fdca976e2adeb30d20c": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_85d9ede0e7064521bfe7a92d54df211f",
- "IPY_MODEL_740857a56a134b2dba327dc8bdbc974a",
- "IPY_MODEL_725534abda3c462188fa27a978bfb59d"
- ],
- "layout": "IPY_MODEL_c61150dc57cb4279b38064e1bd82b3bc"
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "c333e3f23b8f4df6a7fd9f7a000a54ab",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "special_tokens_map.json: 0%| | 0.00/345 [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
+ }
+ ],
+ "source": [
+ "model, tokenizer = FastLanguageModel.from_pretrained(\n",
+ " model_name = \"unsloth/Meta-Llama-3.1-8B\",\n",
+ " max_seq_length = max_seq_length,\n",
+ " dtype = dtype,\n",
+ " load_in_4bit = load_in_4bit,\n",
+ " token = HF_TOKEN\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "tbcH3PNSwRvy"
+ },
+ "source": [
+ "## ⚙ Add LoRA adapters"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 9,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
},
- "85d9ede0e7064521bfe7a92d54df211f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6472e21298f44e6e9c5a3eefcedc59fc",
- "placeholder": "",
- "style": "IPY_MODEL_adb028ffb1c045f9ade7be249a33b375",
- "value": "generation_config.json: 100%"
- }
+ "id": "BnXS0Trn8md9",
+ "outputId": "88053e64-e9c4-4a07-a67e-c61f5647a6cc"
+ },
+ "outputs": [
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "Unsloth 2024.8 patched 32 layers with 32 QKV layers, 32 O layers and 32 MLP layers.\n"
+ ]
+ }
+ ],
+ "source": [
+ "model = FastLanguageModel.get_peft_model(\n",
+ " model,\n",
+ " r = 16, # Suggested 8, 16, 32, 64, 128\n",
+ " target_modules = [\"q_proj\", \"k_proj\", \"v_proj\", \"o_proj\",\n",
+ " \"gate_proj\", \"up_proj\", \"down_proj\",],\n",
+ " lora_alpha = 16,\n",
+ " lora_dropout = 0, # Supports any, but = 0 is optimized\n",
+ " bias = \"none\", # Supports any, but = \"none\" is optimized\n",
+ " use_gradient_checkpointing = \"unsloth\", # True or \"unsloth\" for very long context\n",
+ " random_state = 3407,\n",
+ " use_rslora = False, # rank stabilized LoRA\n",
+ " loftq_config = None, # LoftQ\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "AoqQQ5n4wcfx"
+ },
+ "source": [
+ "## ⚙ Data preparation"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 10,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/",
+ "height": 145,
+ "referenced_widgets": [
+ "ab91405c124b4e408a8c9c2ffa694b2d",
+ "f92d30153e6c45b9a96ef197d5ece260",
+ "3ec8fbee51cf4b72a012cd02983949af",
+ "3f40bad05ed54d87a09eedce846206ed",
+ "5c4d83ae4b364cd9aacc1fa045701c82",
+ "a7798e04e7594d92823d43f231c5111f",
+ "9af66428bb2a425bb8f941c3ec9b55c2",
+ "0d35f1bb1aba4da984da6ad329509f37",
+ "d1d0801d133543b599a2f11a797b8790",
+ "6863f375ef34480fa548b0cf5af2f8c7",
+ "6e9e00717fb642b19d38722068990c73",
+ "cd56ecc4022a4d35a30213398727bd64",
+ "fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
+ "36d90687dc6a48d89768a3e7189e3682",
+ "8697510e2823440aa123010521e81514",
+ "166dde4c3cd34bbfb142b1f9b81a0c66",
+ "83c3e9399bf647a9b4ac850f5c4ff158",
+ "bfcefbe4f0784408a20f83a36bb908ec",
+ "34e7842721d2471886a65d6038137b1c",
+ "1fc06548939b4fbbaeecb7c624bfc337",
+ "d6fe73f0d574413a82b08c1a59d15215",
+ "39dd3663bd5d45deb2cf0b1f880f779f",
+ "240d39dca5e84fb5887ec67d5ff0090f",
+ "0ce31aa585754163b42c6d42f879c8d0",
+ "37975894b2a349d680b50199efc7bc0f",
+ "e12ed5cc35634119b1b6ee7a897fbe24",
+ "6bfd07f4b40b44afbd8edff387ce6d6a",
+ "77777092be6745c693d3c04d81b4dac0",
+ "c86e67b875014a52b61b9bf6e31d1bf8",
+ "a4acc044177d4893a7acb5d4907958ed",
+ "fe1ba7f14bee4303bc5049d761e53d84",
+ "f183b5acd5ea4a718841472ec865d7ac",
+ "d541d6e587cb4231bd620426524479d6",
+ "1809780cf0bf4855bd66beb0e67d3581",
+ "9ef418a30d2643e6b4822f6c877a9b88",
+ "5e0458baf7c74f43b93e7fdeba1805e3",
+ "f50eebbd811e4f1a859bcdef599f6573",
+ "54c7c267468d461790bb2678fe99dbee",
+ "ca9dc9e749b84defab5ef912ceb0df47",
+ "26dfc0e6e9a64e62b1021492ae481658",
+ "69e0d359a8544795b122bfe9ab3c9b03",
+ "ab1610792a52438c8a19d0ae88380c40",
+ "5460a7099f1c466ea6d2feb2d69dd7ef",
+ "652e3281bb4c40ebbf016d53ad680a51"
+ ]
},
- "740857a56a134b2dba327dc8bdbc974a": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a1a32114bd2d4cb8882afebf56d64d41",
- "max": 230,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_8c6ee78067c0490292ee7845dd886a1b",
- "value": 230
- }
- },
- "725534abda3c462188fa27a978bfb59d": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6d28c2bb843a48908827b362ad094183",
- "placeholder": "",
- "style": "IPY_MODEL_41d5a34977dd4488ba5b88c308301fe6",
- "value": " 230/230 [00:00<00:00, 20.6kB/s]"
- }
+ "id": "z9XFfoIz8sTI",
+ "outputId": "27ec45bf-dd8c-4282-d156-e71acb2b7292"
+ },
+ "outputs": [
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "ab91405c124b4e408a8c9c2ffa694b2d",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "Downloading readme: 0%| | 0.00/11.6k [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "c61150dc57cb4279b38064e1bd82b3bc": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "cd56ecc4022a4d35a30213398727bd64",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "Downloading data: 0%| | 0.00/44.3M [00:00, ?B/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "6472e21298f44e6e9c5a3eefcedc59fc": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "240d39dca5e84fb5887ec67d5ff0090f",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "Generating train split: 0%| | 0/51760 [00:00, ? examples/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "adb028ffb1c045f9ade7be249a33b375": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "1809780cf0bf4855bd66beb0e67d3581",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "Map: 0%| | 0/51760 [00:00, ? examples/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
+ }
+ ],
+ "source": [
+ "alpaca_prompt = \"\"\"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n",
+ "\n",
+ "### Instruction:\n",
+ "{}\n",
+ "\n",
+ "### Input:\n",
+ "{}\n",
+ "\n",
+ "### Response:\n",
+ "{}\"\"\"\n",
+ "\n",
+ "EOS_TOKEN = tokenizer.eos_token # add EOS_TOKEN\n",
+ "def formatting_prompts_func(examples):\n",
+ " instructions = examples[\"instruction\"]\n",
+ " inputs = examples[\"input\"]\n",
+ " outputs = examples[\"output\"]\n",
+ " texts = []\n",
+ " for instruction, input, output in zip(instructions, inputs, outputs):\n",
+ " # add EOS_TOKEN, otherwise your generation will go on forever\n",
+ " text = alpaca_prompt.format(instruction, input, output) + EOS_TOKEN\n",
+ " texts.append(text)\n",
+ " return { \"text\" : texts, }\n",
+ "pass\n",
+ "\n",
+ "from datasets import load_dataset\n",
+ "dataset = load_dataset(\"yahma/alpaca-cleaned\", split = \"train\")\n",
+ "dataset = dataset.map(formatting_prompts_func, batched = True,)"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "wljLKG7LwiHE"
+ },
+ "source": [
+ "## ⚙ Training"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 11,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/",
+ "height": 67,
+ "referenced_widgets": [
+ "1bfcc96f38624d9d83dda4cbd60dfcc4",
+ "f5612cfdb6b148c1a15e7d496012b9a5",
+ "76123b2e2fa54ec0a18b306d83b4dc18",
+ "abbbcf602b354fac8802267913a1a4a1",
+ "3d8baf731e0f451980878320adfb5537",
+ "6380d028f75d4508b71fad6deac85a41",
+ "104d453482324cbebc57f44f73173fca",
+ "34949a7431564226a4fc3f0f0edfa740",
+ "542cef32d1874a5f9a830217a123057a",
+ "cd573225788b421e92fdda6d9cb5d5d1",
+ "962212c7cd9c4f6b8dad4b503780a1a9"
+ ]
},
- "a1a32114bd2d4cb8882afebf56d64d41": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "8c6ee78067c0490292ee7845dd886a1b": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "6d28c2bb843a48908827b362ad094183": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "41d5a34977dd4488ba5b88c308301fe6": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
+ "id": "Ra62g_5f8vYo",
+ "outputId": "bfb55505-0a0e-4aa3-9b49-f995513b33a2"
+ },
+ "outputs": [
+ {
+ "data": {
+ "application/vnd.jupyter.widget-view+json": {
+ "model_id": "1bfcc96f38624d9d83dda4cbd60dfcc4",
+ "version_major": 2,
+ "version_minor": 0
+ },
+ "text/plain": [
+ "Map (num_proc=2): 0%| | 0/51760 [00:00, ? examples/s]"
+ ]
+ },
+ "metadata": {},
+ "output_type": "display_data"
},
- "d0ce911bc4c845e88168f32781cf7923": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_718199222cfc412e9e1ee351b45c9db6",
- "IPY_MODEL_dcde9dfd56de417fb2057b3ccc9725f1",
- "IPY_MODEL_c749164ce965417db5411d00f9401623"
- ],
- "layout": "IPY_MODEL_61a290abfcb14c4f8b6cd9e453600bc5"
- }
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "max_steps is given, it will override any value given in num_train_epochs\n"
+ ]
+ }
+ ],
+ "source": [
+ "from trl import SFTTrainer\n",
+ "from transformers import TrainingArguments\n",
+ "from unsloth import is_bfloat16_supported\n",
+ "\n",
+ "trainer = SFTTrainer(\n",
+ " model = model,\n",
+ " tokenizer = tokenizer,\n",
+ " train_dataset = dataset,\n",
+ " dataset_text_field = \"text\",\n",
+ " max_seq_length = max_seq_length,\n",
+ " dataset_num_proc = 2,\n",
+ " packing = False, # Can make training 5x faster for short sequences.\n",
+ " args = TrainingArguments(\n",
+ " per_device_train_batch_size = 2,\n",
+ " gradient_accumulation_steps = 4,\n",
+ " warmup_steps = 5,\n",
+ " # num_train_epochs = 1, # Set this for 1 full training run.\n",
+ " max_steps = 60,\n",
+ " learning_rate = 2e-4,\n",
+ " fp16 = not is_bfloat16_supported(),\n",
+ " bf16 = is_bfloat16_supported(),\n",
+ " logging_steps = 1,\n",
+ " optim = \"adamw_8bit\",\n",
+ " weight_decay = 0.01,\n",
+ " lr_scheduler_type = \"linear\",\n",
+ " seed = 3407,\n",
+ " output_dir = \"outputs\",\n",
+ " ),\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 12,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/",
+ "height": 1000
},
- "718199222cfc412e9e1ee351b45c9db6": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_7cce70a2af1a4fc2a04658264c37b0c7",
- "placeholder": "",
- "style": "IPY_MODEL_096fb8d33b804a02b12bc085fa3c372f",
- "value": "tokenizer_config.json: 100%"
- }
- },
- "dcde9dfd56de417fb2057b3ccc9725f1": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_c6345c375f474aaf88b7b73e662b20f2",
- "max": 50570,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_b68053e4411747b0b12ab63d309aa765",
- "value": 50570
- }
- },
- "c749164ce965417db5411d00f9401623": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_76664f2e75744dbf875de8fd4fdbe2d4",
- "placeholder": "",
- "style": "IPY_MODEL_05891b9099734df397dc8bb7396ef3e2",
- "value": " 50.6k/50.6k [00:00<00:00, 236kB/s]"
- }
- },
- "61a290abfcb14c4f8b6cd9e453600bc5": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "7cce70a2af1a4fc2a04658264c37b0c7": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "096fb8d33b804a02b12bc085fa3c372f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "c6345c375f474aaf88b7b73e662b20f2": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "b68053e4411747b0b12ab63d309aa765": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "76664f2e75744dbf875de8fd4fdbe2d4": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "05891b9099734df397dc8bb7396ef3e2": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "01f7a6c4eec04fc9b063add3719c0777": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_e4f2492f92a74794aec11bf7e2e8d554",
- "IPY_MODEL_cba30ecbcad14542b3acc82e4396275c",
- "IPY_MODEL_249258f9a1764d40a440a5846227059b"
- ],
- "layout": "IPY_MODEL_53c336629bca4e6cacfe69ffa316890a"
- }
- },
- "e4f2492f92a74794aec11bf7e2e8d554": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_b7c0946c9f1b40d5b469e28149808565",
- "placeholder": "",
- "style": "IPY_MODEL_1bc726cc52084737a0269d01f9df6a16",
- "value": "tokenizer.json: 100%"
- }
- },
- "cba30ecbcad14542b3acc82e4396275c": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_51b3596a668742acab22e5a6659d2c55",
- "max": 9085657,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_30d75e70fb6e4c4db53137804d55295d",
- "value": 9085657
- }
- },
- "249258f9a1764d40a440a5846227059b": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_2c6dd56cf2f740cca814d5c6fa504d93",
- "placeholder": "",
- "style": "IPY_MODEL_55e22c9351de46c4a192008dde020393",
- "value": " 9.09M/9.09M [00:01<00:00, 6.98MB/s]"
- }
- },
- "53c336629bca4e6cacfe69ffa316890a": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "b7c0946c9f1b40d5b469e28149808565": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1bc726cc52084737a0269d01f9df6a16": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "51b3596a668742acab22e5a6659d2c55": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "30d75e70fb6e4c4db53137804d55295d": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "2c6dd56cf2f740cca814d5c6fa504d93": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "55e22c9351de46c4a192008dde020393": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "c333e3f23b8f4df6a7fd9f7a000a54ab": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_903f318a80d849098b11451a8c855d50",
- "IPY_MODEL_6b7e50badd5b4c7b979d20d7ffd1b132",
- "IPY_MODEL_d83d854a33ec4adabc9bef9c2b699a0b"
- ],
- "layout": "IPY_MODEL_1be75bb752214bdc8a3ef6e6976f3312"
- }
- },
- "903f318a80d849098b11451a8c855d50": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_d03ff62190934187be5be033a6928545",
- "placeholder": "",
- "style": "IPY_MODEL_1ec6fa8d08e642059f1d673f6260169f",
- "value": "special_tokens_map.json: 100%"
- }
- },
- "6b7e50badd5b4c7b979d20d7ffd1b132": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_23351e3d7e8741a1adad4c1292dab841",
- "max": 345,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_33c4f5595d924cd3a6d8c1575815a5c8",
- "value": 345
- }
- },
- "d83d854a33ec4adabc9bef9c2b699a0b": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6c4cf50fd9ee484396e8272591d21ee9",
- "placeholder": "",
- "style": "IPY_MODEL_124e70720a374644a6d360926059911f",
- "value": " 345/345 [00:00<00:00, 26.3kB/s]"
- }
- },
- "1be75bb752214bdc8a3ef6e6976f3312": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d03ff62190934187be5be033a6928545": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1ec6fa8d08e642059f1d673f6260169f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "23351e3d7e8741a1adad4c1292dab841": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "33c4f5595d924cd3a6d8c1575815a5c8": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "6c4cf50fd9ee484396e8272591d21ee9": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "124e70720a374644a6d360926059911f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "ab91405c124b4e408a8c9c2ffa694b2d": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_f92d30153e6c45b9a96ef197d5ece260",
- "IPY_MODEL_3ec8fbee51cf4b72a012cd02983949af",
- "IPY_MODEL_3f40bad05ed54d87a09eedce846206ed"
- ],
- "layout": "IPY_MODEL_5c4d83ae4b364cd9aacc1fa045701c82"
- }
- },
- "f92d30153e6c45b9a96ef197d5ece260": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a7798e04e7594d92823d43f231c5111f",
- "placeholder": "",
- "style": "IPY_MODEL_9af66428bb2a425bb8f941c3ec9b55c2",
- "value": "Downloading readme: 100%"
- }
- },
- "3ec8fbee51cf4b72a012cd02983949af": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_0d35f1bb1aba4da984da6ad329509f37",
- "max": 11610,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_d1d0801d133543b599a2f11a797b8790",
- "value": 11610
- }
- },
- "3f40bad05ed54d87a09eedce846206ed": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6863f375ef34480fa548b0cf5af2f8c7",
- "placeholder": "",
- "style": "IPY_MODEL_6e9e00717fb642b19d38722068990c73",
- "value": " 11.6k/11.6k [00:00<00:00, 16.5kB/s]"
- }
- },
- "5c4d83ae4b364cd9aacc1fa045701c82": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "a7798e04e7594d92823d43f231c5111f": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "9af66428bb2a425bb8f941c3ec9b55c2": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "0d35f1bb1aba4da984da6ad329509f37": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d1d0801d133543b599a2f11a797b8790": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "6863f375ef34480fa548b0cf5af2f8c7": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6e9e00717fb642b19d38722068990c73": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "cd56ecc4022a4d35a30213398727bd64": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
- "IPY_MODEL_36d90687dc6a48d89768a3e7189e3682",
- "IPY_MODEL_8697510e2823440aa123010521e81514"
- ],
- "layout": "IPY_MODEL_166dde4c3cd34bbfb142b1f9b81a0c66"
- }
- },
- "fb55ed1c28fe4ebe9e9b6b0a7689c3a8": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_83c3e9399bf647a9b4ac850f5c4ff158",
- "placeholder": "",
- "style": "IPY_MODEL_bfcefbe4f0784408a20f83a36bb908ec",
- "value": "Downloading data: 100%"
- }
- },
- "36d90687dc6a48d89768a3e7189e3682": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_34e7842721d2471886a65d6038137b1c",
- "max": 44307561,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_1fc06548939b4fbbaeecb7c624bfc337",
- "value": 44307561
- }
- },
- "8697510e2823440aa123010521e81514": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_d6fe73f0d574413a82b08c1a59d15215",
- "placeholder": "",
- "style": "IPY_MODEL_39dd3663bd5d45deb2cf0b1f880f779f",
- "value": " 44.3M/44.3M [00:00<00:00, 35.4MB/s]"
- }
- },
- "166dde4c3cd34bbfb142b1f9b81a0c66": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "83c3e9399bf647a9b4ac850f5c4ff158": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "bfcefbe4f0784408a20f83a36bb908ec": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "34e7842721d2471886a65d6038137b1c": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1fc06548939b4fbbaeecb7c624bfc337": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "d6fe73f0d574413a82b08c1a59d15215": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "39dd3663bd5d45deb2cf0b1f880f779f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "240d39dca5e84fb5887ec67d5ff0090f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_0ce31aa585754163b42c6d42f879c8d0",
- "IPY_MODEL_37975894b2a349d680b50199efc7bc0f",
- "IPY_MODEL_e12ed5cc35634119b1b6ee7a897fbe24"
- ],
- "layout": "IPY_MODEL_6bfd07f4b40b44afbd8edff387ce6d6a"
- }
- },
- "0ce31aa585754163b42c6d42f879c8d0": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_77777092be6745c693d3c04d81b4dac0",
- "placeholder": "",
- "style": "IPY_MODEL_c86e67b875014a52b61b9bf6e31d1bf8",
- "value": "Generating train split: 100%"
- }
- },
- "37975894b2a349d680b50199efc7bc0f": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a4acc044177d4893a7acb5d4907958ed",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_fe1ba7f14bee4303bc5049d761e53d84",
- "value": 51760
- }
- },
- "e12ed5cc35634119b1b6ee7a897fbe24": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_f183b5acd5ea4a718841472ec865d7ac",
- "placeholder": "",
- "style": "IPY_MODEL_d541d6e587cb4231bd620426524479d6",
- "value": " 51760/51760 [00:00<00:00, 136918.86 examples/s]"
- }
- },
- "6bfd07f4b40b44afbd8edff387ce6d6a": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "77777092be6745c693d3c04d81b4dac0": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "c86e67b875014a52b61b9bf6e31d1bf8": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "a4acc044177d4893a7acb5d4907958ed": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "fe1ba7f14bee4303bc5049d761e53d84": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "f183b5acd5ea4a718841472ec865d7ac": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d541d6e587cb4231bd620426524479d6": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "1809780cf0bf4855bd66beb0e67d3581": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_9ef418a30d2643e6b4822f6c877a9b88",
- "IPY_MODEL_5e0458baf7c74f43b93e7fdeba1805e3",
- "IPY_MODEL_f50eebbd811e4f1a859bcdef599f6573"
- ],
- "layout": "IPY_MODEL_54c7c267468d461790bb2678fe99dbee"
- }
- },
- "9ef418a30d2643e6b4822f6c877a9b88": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_ca9dc9e749b84defab5ef912ceb0df47",
- "placeholder": "",
- "style": "IPY_MODEL_26dfc0e6e9a64e62b1021492ae481658",
- "value": "Map: 100%"
- }
- },
- "5e0458baf7c74f43b93e7fdeba1805e3": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_69e0d359a8544795b122bfe9ab3c9b03",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_ab1610792a52438c8a19d0ae88380c40",
- "value": 51760
- }
- },
- "f50eebbd811e4f1a859bcdef599f6573": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_5460a7099f1c466ea6d2feb2d69dd7ef",
- "placeholder": "",
- "style": "IPY_MODEL_652e3281bb4c40ebbf016d53ad680a51",
- "value": " 51760/51760 [00:00<00:00, 113885.51 examples/s]"
- }
- },
- "54c7c267468d461790bb2678fe99dbee": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "ca9dc9e749b84defab5ef912ceb0df47": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "26dfc0e6e9a64e62b1021492ae481658": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "69e0d359a8544795b122bfe9ab3c9b03": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "ab1610792a52438c8a19d0ae88380c40": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "5460a7099f1c466ea6d2feb2d69dd7ef": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "652e3281bb4c40ebbf016d53ad680a51": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "1bfcc96f38624d9d83dda4cbd60dfcc4": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HBoxModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_f5612cfdb6b148c1a15e7d496012b9a5",
- "IPY_MODEL_76123b2e2fa54ec0a18b306d83b4dc18",
- "IPY_MODEL_abbbcf602b354fac8802267913a1a4a1"
- ],
- "layout": "IPY_MODEL_3d8baf731e0f451980878320adfb5537"
- }
- },
- "f5612cfdb6b148c1a15e7d496012b9a5": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6380d028f75d4508b71fad6deac85a41",
- "placeholder": "",
- "style": "IPY_MODEL_104d453482324cbebc57f44f73173fca",
- "value": "Map (num_proc=2): 100%"
- }
- },
- "76123b2e2fa54ec0a18b306d83b4dc18": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "FloatProgressModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_34949a7431564226a4fc3f0f0edfa740",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_542cef32d1874a5f9a830217a123057a",
- "value": 51760
- }
- },
- "abbbcf602b354fac8802267913a1a4a1": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "HTMLModel",
- "model_module_version": "1.5.0",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_cd573225788b421e92fdda6d9cb5d5d1",
- "placeholder": "",
- "style": "IPY_MODEL_962212c7cd9c4f6b8dad4b503780a1a9",
- "value": " 51760/51760 [00:15<00:00, 3478.84 examples/s]"
- }
- },
- "3d8baf731e0f451980878320adfb5537": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6380d028f75d4508b71fad6deac85a41": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "104d453482324cbebc57f44f73173fca": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "34949a7431564226a4fc3f0f0edfa740": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "542cef32d1874a5f9a830217a123057a": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "ProgressStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "cd573225788b421e92fdda6d9cb5d5d1": {
- "model_module": "@jupyter-widgets/base",
- "model_name": "LayoutModel",
- "model_module_version": "1.2.0",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "962212c7cd9c4f6b8dad4b503780a1a9": {
- "model_module": "@jupyter-widgets/controls",
- "model_name": "DescriptionStyleModel",
- "model_module_version": "1.5.0",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- }
- }
- }
- },
- "cells": [
- {
- "cell_type": "markdown",
- "source": [
- "\n",
- " \n",
- ""
- ],
- "metadata": {
- "id": "6RF2EQaKDoxr"
- }
- },
- {
- "cell_type": "markdown",
- "source": [
- "# Comet and unsloth\n",
- "\n",
- "[Comet](https://www.comet.com/site/?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) is an MLOps platform designed to help data scientists and teams build better models faster! Comet provides tooling to track, explain, manage, and monitor your models in a single place! It works with Jupyter notebooks and scripts and-- most importantly--it's 100% free to get started!\n",
- "\n",
- "[unsloth](https://github.com/unslothai/unsloth) dramatically improves the speed and efficiency of LLM fine-tuning for models including Llama, Phi-3, Gemma, Mistral, and more. For a full listed of 100+ supported unsloth models, [see here](https://huggingface.co/unsloth).\n",
- "\n",
- "Instrument your torchtune training runs with Comet to start managing experiments with efficiency, reproducibility, and collaboration in mind.\n",
- "\n",
- "Find more information about [our integration with torchtune here](https://www.comet.com/docs/v2/integrations/third-party-tools/unsloth?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) or [learn about our other integrations here](https://www.comet.com/docs/v2/integrations?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth)."
- ],
- "metadata": {
- "id": "KDjO5WbeDtp0"
- }
- },
- {
- "cell_type": "markdown",
- "source": [
- "## ⚙ Install and import dependencies"
- ],
- "metadata": {
- "id": "f0fhpJZgsYII"
- }
- },
- {
- "cell_type": "code",
- "execution_count": 1,
- "metadata": {
- "id": "5zTSjY9r4cGc"
- },
- "outputs": [],
- "source": [
- "%%capture\n",
- "!pip install comet_ml \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"\n",
- "\n",
- "# check torch version for Xformers (2.3 -> 0.0.27)\n",
- "from torch import __version__; from packaging.version import Version as V\n",
- "xformers = \"xformers==0.0.27\" if V(__version__) < V(\"2.4.0\") else \"xformers\"\n",
- "!pip install --no-deps {xformers} trl peft accelerate bitsandbytes triton"
- ]
- },
- {
- "cell_type": "code",
- "source": [
- "import comet_ml\n",
- "import os\n",
- "from google.colab import userdata\n",
- "\n",
- "# Log in to Comet\n",
- "COMET_API_KEY = userdata.get('COMET_API_KEY')\n",
- "comet_ml.login(api_key = COMET_API_KEY)\n",
- "# Create experiment object\n",
- "exp = comet_ml.Experiment(project_name=\"comet-example-unsloth\", workspace=\"examples\")\n",
- "\n",
- "# Set Hugging Face token (for dataset retrieval)\n",
- "HF_TOKEN = userdata.get('HF_TOKEN')\n",
- "os.environ[\"HF_TOKEN\"] = HF_TOKEN"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "p8GTqJOW88fI",
- "outputId": "1c7f76ee-a9ff-418d-8067-4cb357f938e5"
- },
- "execution_count": 5,
- "outputs": [
- {
- "output_type": "stream",
- "name": "stderr",
- "text": [
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Valid Comet API Key saved in /root/.comet.config (set COMET_CONFIG to change where it is saved).\n",
- "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n",
- "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m As you are running in a Jupyter environment, you will need to call `experiment.end()` when finished to ensure all metrics and code are logged before exiting.\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Couldn't find a Git repository in '/content' nor in any parent directory. Set `COMET_GIT_DIRECTORY` if your Git Repository is elsewhere.\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Experiment is live on comet.com https://www.comet.com/examples/comet-example-unsloth/cb280e1a2ac942cbab2343349d80282f\n",
- "\n"
- ]
- }
- ]
- },
- {
- "cell_type": "markdown",
- "source": [
- "## ⚙ Download model"
- ],
- "metadata": {
- "id": "5bfxeJqKwpKj"
- }
- },
- {
- "cell_type": "code",
- "source": [
- "from unsloth import FastLanguageModel\n",
- "import torch\n",
- "\n",
- "max_seq_length = 2048\n",
- "dtype = None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+\n",
- "load_in_4bit = True"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "bZei5Dvw7_kM",
- "outputId": "62f7a016-6c2b-4f76-ebf2-f2051b907932"
- },
- "execution_count": 7,
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.\n"
- ]
- }
- ]
- },
- {
- "cell_type": "markdown",
- "source": [
- "Find the full list of [100+ supported unsloth models here](https://huggingface.co/unsloth). For a full list of supported 4-bit models see [here](https://huggingface.co/collections/unsloth/load-4bit-models-4x-faster-659042e3a41c3cbad582e734)"
- ],
- "metadata": {
- "id": "NG--Sbr1n5Cv"
- }
- },
- {
- "cell_type": "code",
- "source": [
- "model, tokenizer = FastLanguageModel.from_pretrained(\n",
- " model_name = \"unsloth/Meta-Llama-3.1-8B\",\n",
- " max_seq_length = max_seq_length,\n",
- " dtype = dtype,\n",
- " load_in_4bit = load_in_4bit,\n",
- " token = HF_TOKEN\n",
- ")"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 284,
- "referenced_widgets": [
- "77caf8ce2875463f86cb7e5b0e02d10f",
- "986a13810ebf4e11ac2d636013efc948",
- "0902e41471fb40f7a7fb807397d1e780",
- "ec2615ff447e4a72a359143122a58887",
- "bb0052241a574fdbb6ca8c942d99e1eb",
- "1aff4958133640838a260a881e84bd66",
- "b920a8e467d64b629189d1c76541cd17",
- "0fe442a6a70948a4a9b703352148d780",
- "5bf1ce41bf6c4b389bac20669bd64e3b",
- "7d06574680cb41b9bfac0c3e47916d6e",
- "b6cd239b54c345feb11143f2364fcfaf",
- "e38214ae388e4fdca976e2adeb30d20c",
- "85d9ede0e7064521bfe7a92d54df211f",
- "740857a56a134b2dba327dc8bdbc974a",
- "725534abda3c462188fa27a978bfb59d",
- "c61150dc57cb4279b38064e1bd82b3bc",
- "6472e21298f44e6e9c5a3eefcedc59fc",
- "adb028ffb1c045f9ade7be249a33b375",
- "a1a32114bd2d4cb8882afebf56d64d41",
- "8c6ee78067c0490292ee7845dd886a1b",
- "6d28c2bb843a48908827b362ad094183",
- "41d5a34977dd4488ba5b88c308301fe6",
- "d0ce911bc4c845e88168f32781cf7923",
- "718199222cfc412e9e1ee351b45c9db6",
- "dcde9dfd56de417fb2057b3ccc9725f1",
- "c749164ce965417db5411d00f9401623",
- "61a290abfcb14c4f8b6cd9e453600bc5",
- "7cce70a2af1a4fc2a04658264c37b0c7",
- "096fb8d33b804a02b12bc085fa3c372f",
- "c6345c375f474aaf88b7b73e662b20f2",
- "b68053e4411747b0b12ab63d309aa765",
- "76664f2e75744dbf875de8fd4fdbe2d4",
- "05891b9099734df397dc8bb7396ef3e2",
- "01f7a6c4eec04fc9b063add3719c0777",
- "e4f2492f92a74794aec11bf7e2e8d554",
- "cba30ecbcad14542b3acc82e4396275c",
- "249258f9a1764d40a440a5846227059b",
- "53c336629bca4e6cacfe69ffa316890a",
- "b7c0946c9f1b40d5b469e28149808565",
- "1bc726cc52084737a0269d01f9df6a16",
- "51b3596a668742acab22e5a6659d2c55",
- "30d75e70fb6e4c4db53137804d55295d",
- "2c6dd56cf2f740cca814d5c6fa504d93",
- "55e22c9351de46c4a192008dde020393",
- "c333e3f23b8f4df6a7fd9f7a000a54ab",
- "903f318a80d849098b11451a8c855d50",
- "6b7e50badd5b4c7b979d20d7ffd1b132",
- "d83d854a33ec4adabc9bef9c2b699a0b",
- "1be75bb752214bdc8a3ef6e6976f3312",
- "d03ff62190934187be5be033a6928545",
- "1ec6fa8d08e642059f1d673f6260169f",
- "23351e3d7e8741a1adad4c1292dab841",
- "33c4f5595d924cd3a6d8c1575815a5c8",
- "6c4cf50fd9ee484396e8272591d21ee9",
- "124e70720a374644a6d360926059911f"
- ]
- },
- "id": "L09m09Xs8Y2e",
- "outputId": "4c67ce27-09cd-44dd-e987-7168a29ee335"
- },
- "execution_count": 8,
- "outputs": [
- {
- "output_type": "stream",
- "name": "stdout",
- "text": [
- "==((====))== Unsloth 2024.8: Fast Llama patching. Transformers = 4.44.2.\n",
- " \\\\ /| GPU: NVIDIA A100-SXM4-40GB. Max memory: 39.564 GB. Platform = Linux.\n",
- "O^O/ \\_/ \\ Pytorch: 2.4.0+cu121. CUDA = 8.0. CUDA Toolkit = 12.1.\n",
- "\\ / Bfloat16 = TRUE. FA [Xformers = 0.0.27.post2. FA2 = False]\n",
- " \"-____-\" Free Apache license: http://github.com/unslothai/unsloth\n",
- "Unsloth: Fast downloading is enabled - ignore downloading bars which are red colored!\n"
- ]
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "model.safetensors: 0%| | 0.00/5.70G [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "77caf8ce2875463f86cb7e5b0e02d10f"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "generation_config.json: 0%| | 0.00/230 [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "e38214ae388e4fdca976e2adeb30d20c"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "tokenizer_config.json: 0%| | 0.00/50.6k [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "d0ce911bc4c845e88168f32781cf7923"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "tokenizer.json: 0%| | 0.00/9.09M [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "01f7a6c4eec04fc9b063add3719c0777"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "special_tokens_map.json: 0%| | 0.00/345 [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "c333e3f23b8f4df6a7fd9f7a000a54ab"
- }
- },
- "metadata": {}
- }
- ]
- },
- {
- "cell_type": "markdown",
- "source": [
- "## ⚙ Add LoRA adapters"
- ],
- "metadata": {
- "id": "tbcH3PNSwRvy"
- }
- },
- {
- "cell_type": "code",
- "source": [
- "model = FastLanguageModel.get_peft_model(\n",
- " model,\n",
- " r = 16, # Suggested 8, 16, 32, 64, 128\n",
- " target_modules = [\"q_proj\", \"k_proj\", \"v_proj\", \"o_proj\",\n",
- " \"gate_proj\", \"up_proj\", \"down_proj\",],\n",
- " lora_alpha = 16,\n",
- " lora_dropout = 0, # Supports any, but = 0 is optimized\n",
- " bias = \"none\", # Supports any, but = \"none\" is optimized\n",
- " use_gradient_checkpointing = \"unsloth\", # True or \"unsloth\" for very long context\n",
- " random_state = 3407,\n",
- " use_rslora = False, # rank stabilized LoRA\n",
- " loftq_config = None, # LoftQ\n",
- ")"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "BnXS0Trn8md9",
- "outputId": "88053e64-e9c4-4a07-a67e-c61f5647a6cc"
- },
- "execution_count": 9,
- "outputs": [
- {
- "output_type": "stream",
- "name": "stderr",
- "text": [
- "Unsloth 2024.8 patched 32 layers with 32 QKV layers, 32 O layers and 32 MLP layers.\n"
- ]
- }
- ]
- },
- {
- "cell_type": "markdown",
- "source": [
- "## ⚙ Data preparation"
- ],
- "metadata": {
- "id": "AoqQQ5n4wcfx"
- }
- },
- {
- "cell_type": "code",
- "source": [
- "alpaca_prompt = \"\"\"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n",
- "\n",
- "### Instruction:\n",
- "{}\n",
- "\n",
- "### Input:\n",
- "{}\n",
- "\n",
- "### Response:\n",
- "{}\"\"\"\n",
- "\n",
- "EOS_TOKEN = tokenizer.eos_token # add EOS_TOKEN\n",
- "def formatting_prompts_func(examples):\n",
- " instructions = examples[\"instruction\"]\n",
- " inputs = examples[\"input\"]\n",
- " outputs = examples[\"output\"]\n",
- " texts = []\n",
- " for instruction, input, output in zip(instructions, inputs, outputs):\n",
- " # add EOS_TOKEN, otherwise your generation will go on forever\n",
- " text = alpaca_prompt.format(instruction, input, output) + EOS_TOKEN\n",
- " texts.append(text)\n",
- " return { \"text\" : texts, }\n",
- "pass\n",
- "\n",
- "from datasets import load_dataset\n",
- "dataset = load_dataset(\"yahma/alpaca-cleaned\", split = \"train\")\n",
- "dataset = dataset.map(formatting_prompts_func, batched = True,)"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 145,
- "referenced_widgets": [
- "ab91405c124b4e408a8c9c2ffa694b2d",
- "f92d30153e6c45b9a96ef197d5ece260",
- "3ec8fbee51cf4b72a012cd02983949af",
- "3f40bad05ed54d87a09eedce846206ed",
- "5c4d83ae4b364cd9aacc1fa045701c82",
- "a7798e04e7594d92823d43f231c5111f",
- "9af66428bb2a425bb8f941c3ec9b55c2",
- "0d35f1bb1aba4da984da6ad329509f37",
- "d1d0801d133543b599a2f11a797b8790",
- "6863f375ef34480fa548b0cf5af2f8c7",
- "6e9e00717fb642b19d38722068990c73",
- "cd56ecc4022a4d35a30213398727bd64",
- "fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
- "36d90687dc6a48d89768a3e7189e3682",
- "8697510e2823440aa123010521e81514",
- "166dde4c3cd34bbfb142b1f9b81a0c66",
- "83c3e9399bf647a9b4ac850f5c4ff158",
- "bfcefbe4f0784408a20f83a36bb908ec",
- "34e7842721d2471886a65d6038137b1c",
- "1fc06548939b4fbbaeecb7c624bfc337",
- "d6fe73f0d574413a82b08c1a59d15215",
- "39dd3663bd5d45deb2cf0b1f880f779f",
- "240d39dca5e84fb5887ec67d5ff0090f",
- "0ce31aa585754163b42c6d42f879c8d0",
- "37975894b2a349d680b50199efc7bc0f",
- "e12ed5cc35634119b1b6ee7a897fbe24",
- "6bfd07f4b40b44afbd8edff387ce6d6a",
- "77777092be6745c693d3c04d81b4dac0",
- "c86e67b875014a52b61b9bf6e31d1bf8",
- "a4acc044177d4893a7acb5d4907958ed",
- "fe1ba7f14bee4303bc5049d761e53d84",
- "f183b5acd5ea4a718841472ec865d7ac",
- "d541d6e587cb4231bd620426524479d6",
- "1809780cf0bf4855bd66beb0e67d3581",
- "9ef418a30d2643e6b4822f6c877a9b88",
- "5e0458baf7c74f43b93e7fdeba1805e3",
- "f50eebbd811e4f1a859bcdef599f6573",
- "54c7c267468d461790bb2678fe99dbee",
- "ca9dc9e749b84defab5ef912ceb0df47",
- "26dfc0e6e9a64e62b1021492ae481658",
- "69e0d359a8544795b122bfe9ab3c9b03",
- "ab1610792a52438c8a19d0ae88380c40",
- "5460a7099f1c466ea6d2feb2d69dd7ef",
- "652e3281bb4c40ebbf016d53ad680a51"
- ]
- },
- "id": "z9XFfoIz8sTI",
- "outputId": "27ec45bf-dd8c-4282-d156-e71acb2b7292"
- },
- "execution_count": 10,
- "outputs": [
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "Downloading readme: 0%| | 0.00/11.6k [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "ab91405c124b4e408a8c9c2ffa694b2d"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "Downloading data: 0%| | 0.00/44.3M [00:00, ?B/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "cd56ecc4022a4d35a30213398727bd64"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "Generating train split: 0%| | 0/51760 [00:00, ? examples/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "240d39dca5e84fb5887ec67d5ff0090f"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "Map: 0%| | 0/51760 [00:00, ? examples/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "1809780cf0bf4855bd66beb0e67d3581"
- }
- },
- "metadata": {}
- }
- ]
- },
- {
- "cell_type": "markdown",
- "source": [
- "## ⚙ Training"
- ],
- "metadata": {
- "id": "wljLKG7LwiHE"
- }
- },
- {
- "cell_type": "code",
- "source": [
- "from trl import SFTTrainer\n",
- "from transformers import TrainingArguments\n",
- "from unsloth import is_bfloat16_supported\n",
- "\n",
- "trainer = SFTTrainer(\n",
- " model = model,\n",
- " tokenizer = tokenizer,\n",
- " train_dataset = dataset,\n",
- " dataset_text_field = \"text\",\n",
- " max_seq_length = max_seq_length,\n",
- " dataset_num_proc = 2,\n",
- " packing = False, # Can make training 5x faster for short sequences.\n",
- " args = TrainingArguments(\n",
- " per_device_train_batch_size = 2,\n",
- " gradient_accumulation_steps = 4,\n",
- " warmup_steps = 5,\n",
- " # num_train_epochs = 1, # Set this for 1 full training run.\n",
- " max_steps = 60,\n",
- " learning_rate = 2e-4,\n",
- " fp16 = not is_bfloat16_supported(),\n",
- " bf16 = is_bfloat16_supported(),\n",
- " logging_steps = 1,\n",
- " optim = \"adamw_8bit\",\n",
- " weight_decay = 0.01,\n",
- " lr_scheduler_type = \"linear\",\n",
- " seed = 3407,\n",
- " output_dir = \"outputs\",\n",
- " ),\n",
- ")"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 67,
- "referenced_widgets": [
- "1bfcc96f38624d9d83dda4cbd60dfcc4",
- "f5612cfdb6b148c1a15e7d496012b9a5",
- "76123b2e2fa54ec0a18b306d83b4dc18",
- "abbbcf602b354fac8802267913a1a4a1",
- "3d8baf731e0f451980878320adfb5537",
- "6380d028f75d4508b71fad6deac85a41",
- "104d453482324cbebc57f44f73173fca",
- "34949a7431564226a4fc3f0f0edfa740",
- "542cef32d1874a5f9a830217a123057a",
- "cd573225788b421e92fdda6d9cb5d5d1",
- "962212c7cd9c4f6b8dad4b503780a1a9"
- ]
- },
- "id": "Ra62g_5f8vYo",
- "outputId": "bfb55505-0a0e-4aa3-9b49-f995513b33a2"
- },
- "execution_count": 11,
- "outputs": [
- {
- "output_type": "display_data",
- "data": {
- "text/plain": [
- "Map (num_proc=2): 0%| | 0/51760 [00:00, ? examples/s]"
- ],
- "application/vnd.jupyter.widget-view+json": {
- "version_major": 2,
- "version_minor": 0,
- "model_id": "1bfcc96f38624d9d83dda4cbd60dfcc4"
- }
- },
- "metadata": {}
- },
- {
- "output_type": "stream",
- "name": "stderr",
- "text": [
- "max_steps is given, it will override any value given in num_train_epochs\n"
- ]
- }
- ]
- },
- {
- "cell_type": "code",
- "source": [
- "trainer_stats = trainer.train()"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 1000
- },
- "id": "e5EHhd5zANxX",
- "outputId": "d8381af1-03be-499e-955a-2008cf77081d"
- },
- "execution_count": 12,
- "outputs": [
- {
- "output_type": "stream",
- "name": "stderr",
- "text": [
- "==((====))== Unsloth - 2x faster free finetuning | Num GPUs = 1\n",
- " \\\\ /| Num examples = 51,760 | Num Epochs = 1\n",
- "O^O/ \\_/ \\ Batch size per device = 2 | Gradient Accumulation steps = 4\n",
- "\\ / Total batch size = 8 | Total steps = 60\n",
- " \"-____-\" Number of trainable parameters = 41,943,040\n"
- ]
+ "id": "e5EHhd5zANxX",
+ "outputId": "d8381af1-03be-499e-955a-2008cf77081d"
+ },
+ "outputs": [
+ {
+ "name": "stderr",
+ "output_type": "stream",
+ "text": [
+ "==((====))== Unsloth - 2x faster free finetuning | Num GPUs = 1\n",
+ " \\\\ /| Num examples = 51,760 | Num Epochs = 1\n",
+ "O^O/ \\_/ \\ Batch size per device = 2 | Gradient Accumulation steps = 4\n",
+ "\\ / Total batch size = 8 | Total steps = 60\n",
+ " \"-____-\" Number of trainable parameters = 41,943,040\n"
+ ]
},
{
- "output_type": "display_data",
"data": {
- "text/plain": [
- ""
- ],
"text/html": [
"\n",
" \n",
@@ -4319,17 +872,22 @@
" \n",
" \n",
"
"
+ ],
+ "text/plain": [
+ ""
]
},
- "metadata": {}
+ "metadata": {},
+ "output_type": "display_data"
}
+ ],
+ "source": [
+ "trainer_stats = trainer.train()"
]
},
{
"cell_type": "code",
- "source": [
- "exp.end()"
- ],
+ "execution_count": 13,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/"
@@ -4337,11 +895,10 @@
"id": "8II72I92Bm9s",
"outputId": "6518c199-dff2-4c4a-f821-aefdf595fa2c"
},
- "execution_count": 13,
"outputs": [
{
- "output_type": "stream",
"name": "stderr",
+ "output_type": "stream",
"text": [
"\u001b[1;38;5;39mCOMET INFO:\u001b[0m ---------------------------------------------------------------------------------------\n",
"\u001b[1;38;5;39mCOMET INFO:\u001b[0m Comet.ml Experiment Summary\n",
@@ -4622,19 +1179,42 @@
"\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n"
]
}
+ ],
+ "source": [
+ "exp.end()"
]
},
{
"cell_type": "markdown",
- "source": [
- "## ⚙ Inference"
- ],
"metadata": {
"id": "0IdjUa_0pzPQ"
- }
+ },
+ "source": [
+ "## ⚙ Inference"
+ ]
},
{
"cell_type": "code",
+ "execution_count": 12,
+ "metadata": {
+ "colab": {
+ "base_uri": "https://localhost:8080/"
+ },
+ "id": "76su-yv9Aad0",
+ "outputId": "8b9c2ccd-4597-41fc-d9ef-e665272fae05"
+ },
+ "outputs": [
+ {
+ "data": {
+ "text/plain": [
+ "['<|begin_of_text|>Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nContinue the fibonnaci sequence.\\n\\n### Input:\\n1, 1, 2, 3, 5, 8\\n\\n### Response:\\n13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597, 2584, 4181, 6765, 10946, 17711, 28657, 46368, 75025']"
+ ]
+ },
+ "execution_count": 12,
+ "metadata": {},
+ "output_type": "execute_result"
+ }
+ ],
"source": [
"# alpaca_prompt = Copied from above\n",
"FastLanguageModel.for_inference(model)\n",
@@ -4649,27 +1229,3449 @@
"\n",
"outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)\n",
"tokenizer.batch_decode(outputs)"
- ],
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
+ ]
+ }
+ ],
+ "metadata": {
+ "accelerator": "GPU",
+ "colab": {
+ "gpuType": "A100",
+ "machine_shape": "hm",
+ "provenance": [],
+ "toc_visible": true
+ },
+ "kernelspec": {
+ "display_name": "Python 3",
+ "name": "python3"
+ },
+ "language_info": {
+ "name": "python"
+ },
+ "widgets": {
+ "application/vnd.jupyter.widget-state+json": {
+ "01f7a6c4eec04fc9b063add3719c0777": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_e4f2492f92a74794aec11bf7e2e8d554",
+ "IPY_MODEL_cba30ecbcad14542b3acc82e4396275c",
+ "IPY_MODEL_249258f9a1764d40a440a5846227059b"
+ ],
+ "layout": "IPY_MODEL_53c336629bca4e6cacfe69ffa316890a"
+ }
+ },
+ "05891b9099734df397dc8bb7396ef3e2": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "0902e41471fb40f7a7fb807397d1e780": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "danger",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_0fe442a6a70948a4a9b703352148d780",
+ "max": 5702746390,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_5bf1ce41bf6c4b389bac20669bd64e3b",
+ "value": 5702745847
+ }
+ },
+ "096fb8d33b804a02b12bc085fa3c372f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "0ce31aa585754163b42c6d42f879c8d0": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_77777092be6745c693d3c04d81b4dac0",
+ "placeholder": "",
+ "style": "IPY_MODEL_c86e67b875014a52b61b9bf6e31d1bf8",
+ "value": "Generating train split: 100%"
+ }
+ },
+ "0d35f1bb1aba4da984da6ad329509f37": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "0fe442a6a70948a4a9b703352148d780": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "104d453482324cbebc57f44f73173fca": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "124e70720a374644a6d360926059911f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "166dde4c3cd34bbfb142b1f9b81a0c66": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "1809780cf0bf4855bd66beb0e67d3581": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_9ef418a30d2643e6b4822f6c877a9b88",
+ "IPY_MODEL_5e0458baf7c74f43b93e7fdeba1805e3",
+ "IPY_MODEL_f50eebbd811e4f1a859bcdef599f6573"
+ ],
+ "layout": "IPY_MODEL_54c7c267468d461790bb2678fe99dbee"
+ }
+ },
+ "1aff4958133640838a260a881e84bd66": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "1bc726cc52084737a0269d01f9df6a16": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "1be75bb752214bdc8a3ef6e6976f3312": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "1bfcc96f38624d9d83dda4cbd60dfcc4": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_f5612cfdb6b148c1a15e7d496012b9a5",
+ "IPY_MODEL_76123b2e2fa54ec0a18b306d83b4dc18",
+ "IPY_MODEL_abbbcf602b354fac8802267913a1a4a1"
+ ],
+ "layout": "IPY_MODEL_3d8baf731e0f451980878320adfb5537"
+ }
+ },
+ "1ec6fa8d08e642059f1d673f6260169f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "1fc06548939b4fbbaeecb7c624bfc337": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "23351e3d7e8741a1adad4c1292dab841": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "240d39dca5e84fb5887ec67d5ff0090f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_0ce31aa585754163b42c6d42f879c8d0",
+ "IPY_MODEL_37975894b2a349d680b50199efc7bc0f",
+ "IPY_MODEL_e12ed5cc35634119b1b6ee7a897fbe24"
+ ],
+ "layout": "IPY_MODEL_6bfd07f4b40b44afbd8edff387ce6d6a"
+ }
+ },
+ "249258f9a1764d40a440a5846227059b": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_2c6dd56cf2f740cca814d5c6fa504d93",
+ "placeholder": "",
+ "style": "IPY_MODEL_55e22c9351de46c4a192008dde020393",
+ "value": " 9.09M/9.09M [00:01<00:00, 6.98MB/s]"
+ }
+ },
+ "26dfc0e6e9a64e62b1021492ae481658": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "2c6dd56cf2f740cca814d5c6fa504d93": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "30d75e70fb6e4c4db53137804d55295d": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "33c4f5595d924cd3a6d8c1575815a5c8": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "34949a7431564226a4fc3f0f0edfa740": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "34e7842721d2471886a65d6038137b1c": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "36d90687dc6a48d89768a3e7189e3682": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_34e7842721d2471886a65d6038137b1c",
+ "max": 44307561,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_1fc06548939b4fbbaeecb7c624bfc337",
+ "value": 44307561
+ }
+ },
+ "37975894b2a349d680b50199efc7bc0f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_a4acc044177d4893a7acb5d4907958ed",
+ "max": 51760,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_fe1ba7f14bee4303bc5049d761e53d84",
+ "value": 51760
+ }
+ },
+ "39dd3663bd5d45deb2cf0b1f880f779f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "3d8baf731e0f451980878320adfb5537": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "3ec8fbee51cf4b72a012cd02983949af": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_0d35f1bb1aba4da984da6ad329509f37",
+ "max": 11610,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_d1d0801d133543b599a2f11a797b8790",
+ "value": 11610
+ }
+ },
+ "3f40bad05ed54d87a09eedce846206ed": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_6863f375ef34480fa548b0cf5af2f8c7",
+ "placeholder": "",
+ "style": "IPY_MODEL_6e9e00717fb642b19d38722068990c73",
+ "value": " 11.6k/11.6k [00:00<00:00, 16.5kB/s]"
+ }
+ },
+ "41d5a34977dd4488ba5b88c308301fe6": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "51b3596a668742acab22e5a6659d2c55": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "53c336629bca4e6cacfe69ffa316890a": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "542cef32d1874a5f9a830217a123057a": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "5460a7099f1c466ea6d2feb2d69dd7ef": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "54c7c267468d461790bb2678fe99dbee": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "55e22c9351de46c4a192008dde020393": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "5bf1ce41bf6c4b389bac20669bd64e3b": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "5c4d83ae4b364cd9aacc1fa045701c82": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "5e0458baf7c74f43b93e7fdeba1805e3": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_69e0d359a8544795b122bfe9ab3c9b03",
+ "max": 51760,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_ab1610792a52438c8a19d0ae88380c40",
+ "value": 51760
+ }
+ },
+ "61a290abfcb14c4f8b6cd9e453600bc5": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6380d028f75d4508b71fad6deac85a41": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6472e21298f44e6e9c5a3eefcedc59fc": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "652e3281bb4c40ebbf016d53ad680a51": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "6863f375ef34480fa548b0cf5af2f8c7": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "69e0d359a8544795b122bfe9ab3c9b03": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6b7e50badd5b4c7b979d20d7ffd1b132": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_23351e3d7e8741a1adad4c1292dab841",
+ "max": 345,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_33c4f5595d924cd3a6d8c1575815a5c8",
+ "value": 345
+ }
+ },
+ "6bfd07f4b40b44afbd8edff387ce6d6a": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6c4cf50fd9ee484396e8272591d21ee9": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6d28c2bb843a48908827b362ad094183": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "6e9e00717fb642b19d38722068990c73": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "718199222cfc412e9e1ee351b45c9db6": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_7cce70a2af1a4fc2a04658264c37b0c7",
+ "placeholder": "",
+ "style": "IPY_MODEL_096fb8d33b804a02b12bc085fa3c372f",
+ "value": "tokenizer_config.json: 100%"
+ }
+ },
+ "725534abda3c462188fa27a978bfb59d": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_6d28c2bb843a48908827b362ad094183",
+ "placeholder": "",
+ "style": "IPY_MODEL_41d5a34977dd4488ba5b88c308301fe6",
+ "value": " 230/230 [00:00<00:00, 20.6kB/s]"
+ }
+ },
+ "740857a56a134b2dba327dc8bdbc974a": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_a1a32114bd2d4cb8882afebf56d64d41",
+ "max": 230,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_8c6ee78067c0490292ee7845dd886a1b",
+ "value": 230
+ }
+ },
+ "76123b2e2fa54ec0a18b306d83b4dc18": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_34949a7431564226a4fc3f0f0edfa740",
+ "max": 51760,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_542cef32d1874a5f9a830217a123057a",
+ "value": 51760
+ }
+ },
+ "76664f2e75744dbf875de8fd4fdbe2d4": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "77777092be6745c693d3c04d81b4dac0": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "77caf8ce2875463f86cb7e5b0e02d10f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_986a13810ebf4e11ac2d636013efc948",
+ "IPY_MODEL_0902e41471fb40f7a7fb807397d1e780",
+ "IPY_MODEL_ec2615ff447e4a72a359143122a58887"
+ ],
+ "layout": "IPY_MODEL_bb0052241a574fdbb6ca8c942d99e1eb"
+ }
+ },
+ "7cce70a2af1a4fc2a04658264c37b0c7": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "7d06574680cb41b9bfac0c3e47916d6e": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "83c3e9399bf647a9b4ac850f5c4ff158": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "85d9ede0e7064521bfe7a92d54df211f": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_6472e21298f44e6e9c5a3eefcedc59fc",
+ "placeholder": "",
+ "style": "IPY_MODEL_adb028ffb1c045f9ade7be249a33b375",
+ "value": "generation_config.json: 100%"
+ }
+ },
+ "8697510e2823440aa123010521e81514": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_d6fe73f0d574413a82b08c1a59d15215",
+ "placeholder": "",
+ "style": "IPY_MODEL_39dd3663bd5d45deb2cf0b1f880f779f",
+ "value": " 44.3M/44.3M [00:00<00:00, 35.4MB/s]"
+ }
+ },
+ "8c6ee78067c0490292ee7845dd886a1b": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "903f318a80d849098b11451a8c855d50": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_d03ff62190934187be5be033a6928545",
+ "placeholder": "",
+ "style": "IPY_MODEL_1ec6fa8d08e642059f1d673f6260169f",
+ "value": "special_tokens_map.json: 100%"
+ }
+ },
+ "962212c7cd9c4f6b8dad4b503780a1a9": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "986a13810ebf4e11ac2d636013efc948": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_1aff4958133640838a260a881e84bd66",
+ "placeholder": "",
+ "style": "IPY_MODEL_b920a8e467d64b629189d1c76541cd17",
+ "value": "model.safetensors: 100%"
+ }
+ },
+ "9af66428bb2a425bb8f941c3ec9b55c2": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "9ef418a30d2643e6b4822f6c877a9b88": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_ca9dc9e749b84defab5ef912ceb0df47",
+ "placeholder": "",
+ "style": "IPY_MODEL_26dfc0e6e9a64e62b1021492ae481658",
+ "value": "Map: 100%"
+ }
+ },
+ "a1a32114bd2d4cb8882afebf56d64d41": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "a4acc044177d4893a7acb5d4907958ed": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "a7798e04e7594d92823d43f231c5111f": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "ab1610792a52438c8a19d0ae88380c40": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "ab91405c124b4e408a8c9c2ffa694b2d": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_f92d30153e6c45b9a96ef197d5ece260",
+ "IPY_MODEL_3ec8fbee51cf4b72a012cd02983949af",
+ "IPY_MODEL_3f40bad05ed54d87a09eedce846206ed"
+ ],
+ "layout": "IPY_MODEL_5c4d83ae4b364cd9aacc1fa045701c82"
+ }
+ },
+ "abbbcf602b354fac8802267913a1a4a1": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_cd573225788b421e92fdda6d9cb5d5d1",
+ "placeholder": "",
+ "style": "IPY_MODEL_962212c7cd9c4f6b8dad4b503780a1a9",
+ "value": " 51760/51760 [00:15<00:00, 3478.84 examples/s]"
+ }
+ },
+ "adb028ffb1c045f9ade7be249a33b375": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "b68053e4411747b0b12ab63d309aa765": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "b6cd239b54c345feb11143f2364fcfaf": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "b7c0946c9f1b40d5b469e28149808565": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "b920a8e467d64b629189d1c76541cd17": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "bb0052241a574fdbb6ca8c942d99e1eb": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "bfcefbe4f0784408a20f83a36bb908ec": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "c333e3f23b8f4df6a7fd9f7a000a54ab": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_903f318a80d849098b11451a8c855d50",
+ "IPY_MODEL_6b7e50badd5b4c7b979d20d7ffd1b132",
+ "IPY_MODEL_d83d854a33ec4adabc9bef9c2b699a0b"
+ ],
+ "layout": "IPY_MODEL_1be75bb752214bdc8a3ef6e6976f3312"
+ }
+ },
+ "c61150dc57cb4279b38064e1bd82b3bc": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "c6345c375f474aaf88b7b73e662b20f2": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "c749164ce965417db5411d00f9401623": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_76664f2e75744dbf875de8fd4fdbe2d4",
+ "placeholder": "",
+ "style": "IPY_MODEL_05891b9099734df397dc8bb7396ef3e2",
+ "value": " 50.6k/50.6k [00:00<00:00, 236kB/s]"
+ }
+ },
+ "c86e67b875014a52b61b9bf6e31d1bf8": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "ca9dc9e749b84defab5ef912ceb0df47": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "cba30ecbcad14542b3acc82e4396275c": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_51b3596a668742acab22e5a6659d2c55",
+ "max": 9085657,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_30d75e70fb6e4c4db53137804d55295d",
+ "value": 9085657
+ }
+ },
+ "cd56ecc4022a4d35a30213398727bd64": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
+ "IPY_MODEL_36d90687dc6a48d89768a3e7189e3682",
+ "IPY_MODEL_8697510e2823440aa123010521e81514"
+ ],
+ "layout": "IPY_MODEL_166dde4c3cd34bbfb142b1f9b81a0c66"
+ }
+ },
+ "cd573225788b421e92fdda6d9cb5d5d1": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "d03ff62190934187be5be033a6928545": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "d0ce911bc4c845e88168f32781cf7923": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_718199222cfc412e9e1ee351b45c9db6",
+ "IPY_MODEL_dcde9dfd56de417fb2057b3ccc9725f1",
+ "IPY_MODEL_c749164ce965417db5411d00f9401623"
+ ],
+ "layout": "IPY_MODEL_61a290abfcb14c4f8b6cd9e453600bc5"
+ }
+ },
+ "d1d0801d133543b599a2f11a797b8790": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
+ },
+ "d541d6e587cb4231bd620426524479d6": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "DescriptionStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "DescriptionStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "description_width": ""
+ }
+ },
+ "d6fe73f0d574413a82b08c1a59d15215": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "d83d854a33ec4adabc9bef9c2b699a0b": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_6c4cf50fd9ee484396e8272591d21ee9",
+ "placeholder": "",
+ "style": "IPY_MODEL_124e70720a374644a6d360926059911f",
+ "value": " 345/345 [00:00<00:00, 26.3kB/s]"
+ }
+ },
+ "dcde9dfd56de417fb2057b3ccc9725f1": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "FloatProgressModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "FloatProgressModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "ProgressView",
+ "bar_style": "success",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_c6345c375f474aaf88b7b73e662b20f2",
+ "max": 50570,
+ "min": 0,
+ "orientation": "horizontal",
+ "style": "IPY_MODEL_b68053e4411747b0b12ab63d309aa765",
+ "value": 50570
+ }
},
- "id": "76su-yv9Aad0",
- "outputId": "8b9c2ccd-4597-41fc-d9ef-e665272fae05"
- },
- "execution_count": 12,
- "outputs": [
- {
- "output_type": "execute_result",
- "data": {
- "text/plain": [
- "['<|begin_of_text|>Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nContinue the fibonnaci sequence.\\n\\n### Input:\\n1, 1, 2, 3, 5, 8\\n\\n### Response:\\n13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597, 2584, 4181, 6765, 10946, 17711, 28657, 46368, 75025']"
- ]
- },
- "metadata": {},
- "execution_count": 12
+ "e12ed5cc35634119b1b6ee7a897fbe24": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_f183b5acd5ea4a718841472ec865d7ac",
+ "placeholder": "",
+ "style": "IPY_MODEL_d541d6e587cb4231bd620426524479d6",
+ "value": " 51760/51760 [00:00<00:00, 136918.86 examples/s]"
+ }
+ },
+ "e38214ae388e4fdca976e2adeb30d20c": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HBoxModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HBoxModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HBoxView",
+ "box_style": "",
+ "children": [
+ "IPY_MODEL_85d9ede0e7064521bfe7a92d54df211f",
+ "IPY_MODEL_740857a56a134b2dba327dc8bdbc974a",
+ "IPY_MODEL_725534abda3c462188fa27a978bfb59d"
+ ],
+ "layout": "IPY_MODEL_c61150dc57cb4279b38064e1bd82b3bc"
+ }
+ },
+ "e4f2492f92a74794aec11bf7e2e8d554": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_b7c0946c9f1b40d5b469e28149808565",
+ "placeholder": "",
+ "style": "IPY_MODEL_1bc726cc52084737a0269d01f9df6a16",
+ "value": "tokenizer.json: 100%"
+ }
+ },
+ "ec2615ff447e4a72a359143122a58887": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_7d06574680cb41b9bfac0c3e47916d6e",
+ "placeholder": "",
+ "style": "IPY_MODEL_b6cd239b54c345feb11143f2364fcfaf",
+ "value": " 5.70G/5.70G [00:13<00:00, 698MB/s]"
+ }
+ },
+ "f183b5acd5ea4a718841472ec865d7ac": {
+ "model_module": "@jupyter-widgets/base",
+ "model_module_version": "1.2.0",
+ "model_name": "LayoutModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/base",
+ "_model_module_version": "1.2.0",
+ "_model_name": "LayoutModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "LayoutView",
+ "align_content": null,
+ "align_items": null,
+ "align_self": null,
+ "border": null,
+ "bottom": null,
+ "display": null,
+ "flex": null,
+ "flex_flow": null,
+ "grid_area": null,
+ "grid_auto_columns": null,
+ "grid_auto_flow": null,
+ "grid_auto_rows": null,
+ "grid_column": null,
+ "grid_gap": null,
+ "grid_row": null,
+ "grid_template_areas": null,
+ "grid_template_columns": null,
+ "grid_template_rows": null,
+ "height": null,
+ "justify_content": null,
+ "justify_items": null,
+ "left": null,
+ "margin": null,
+ "max_height": null,
+ "max_width": null,
+ "min_height": null,
+ "min_width": null,
+ "object_fit": null,
+ "object_position": null,
+ "order": null,
+ "overflow": null,
+ "overflow_x": null,
+ "overflow_y": null,
+ "padding": null,
+ "right": null,
+ "top": null,
+ "visibility": null,
+ "width": null
+ }
+ },
+ "f50eebbd811e4f1a859bcdef599f6573": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_5460a7099f1c466ea6d2feb2d69dd7ef",
+ "placeholder": "",
+ "style": "IPY_MODEL_652e3281bb4c40ebbf016d53ad680a51",
+ "value": " 51760/51760 [00:00<00:00, 113885.51 examples/s]"
+ }
+ },
+ "f5612cfdb6b148c1a15e7d496012b9a5": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_6380d028f75d4508b71fad6deac85a41",
+ "placeholder": "",
+ "style": "IPY_MODEL_104d453482324cbebc57f44f73173fca",
+ "value": "Map (num_proc=2): 100%"
+ }
+ },
+ "f92d30153e6c45b9a96ef197d5ece260": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_a7798e04e7594d92823d43f231c5111f",
+ "placeholder": "",
+ "style": "IPY_MODEL_9af66428bb2a425bb8f941c3ec9b55c2",
+ "value": "Downloading readme: 100%"
+ }
+ },
+ "fb55ed1c28fe4ebe9e9b6b0a7689c3a8": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "HTMLModel",
+ "state": {
+ "_dom_classes": [],
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "HTMLModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/controls",
+ "_view_module_version": "1.5.0",
+ "_view_name": "HTMLView",
+ "description": "",
+ "description_tooltip": null,
+ "layout": "IPY_MODEL_83c3e9399bf647a9b4ac850f5c4ff158",
+ "placeholder": "",
+ "style": "IPY_MODEL_bfcefbe4f0784408a20f83a36bb908ec",
+ "value": "Downloading data: 100%"
+ }
+ },
+ "fe1ba7f14bee4303bc5049d761e53d84": {
+ "model_module": "@jupyter-widgets/controls",
+ "model_module_version": "1.5.0",
+ "model_name": "ProgressStyleModel",
+ "state": {
+ "_model_module": "@jupyter-widgets/controls",
+ "_model_module_version": "1.5.0",
+ "_model_name": "ProgressStyleModel",
+ "_view_count": null,
+ "_view_module": "@jupyter-widgets/base",
+ "_view_module_version": "1.2.0",
+ "_view_name": "StyleView",
+ "bar_color": null,
+ "description_width": ""
+ }
}
- ]
+ }
}
- ]
-}
\ No newline at end of file
+ },
+ "nbformat": 4,
+ "nbformat_minor": 0
+}
From 31f2706b6749eee8d150206f61945df6a7e7f537 Mon Sep 17 00:00:00 2001
From: Boris Feld
Date: Tue, 10 Sep 2024 13:50:19 +0200
Subject: [PATCH 3/3] Simplify pip call and remove hardcoded workspace
---
.../unsloth/notebooks/Comet_and_unsloth.ipynb | 5030 ++---------------
1 file changed, 355 insertions(+), 4675 deletions(-)
diff --git a/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb b/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
index 3ef0557..bc60fea 100644
--- a/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
+++ b/integrations/model-training/unsloth/notebooks/Comet_and_unsloth.ipynb
@@ -1,4677 +1,357 @@
{
- "cells": [
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "6RF2EQaKDoxr"
- },
- "source": [
- "\n",
- " \n",
- ""
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "KDjO5WbeDtp0"
- },
- "source": [
- "# Comet and unsloth\n",
- "\n",
- "[Comet](https://www.comet.com/site/?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) is an MLOps platform designed to help data scientists and teams build better models faster! Comet provides tooling to track, explain, manage, and monitor your models in a single place! It works with Jupyter notebooks and scripts and-- most importantly--it's 100% free to get started!\n",
- "\n",
- "[unsloth](https://github.com/unslothai/unsloth) dramatically improves the speed and efficiency of LLM fine-tuning for models including Llama, Phi-3, Gemma, Mistral, and more. For a full listed of 100+ supported unsloth models, [see here](https://huggingface.co/unsloth).\n",
- "\n",
- "Instrument your torchtune training runs with Comet to start managing experiments with efficiency, reproducibility, and collaboration in mind.\n",
- "\n",
- "Find more information about [our integration with torchtune here](https://www.comet.com/docs/v2/integrations/third-party-tools/unsloth?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) or [learn about our other integrations here](https://www.comet.com/docs/v2/integrations?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth)."
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "f0fhpJZgsYII"
- },
- "source": [
- "## ⚙ Install and import dependencies"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 1,
- "metadata": {
- "id": "5zTSjY9r4cGc"
- },
- "outputs": [],
- "source": [
- "%%capture\n",
- "!pip install comet_ml \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\"\n",
- "\n",
- "# check torch version for Xformers (2.3 -> 0.0.27)\n",
- "from torch import __version__; from packaging.version import Version as V\n",
- "xformers = \"xformers==0.0.27\" if V(__version__) < V(\"2.4.0\") else \"xformers\"\n",
- "!pip install --no-deps {xformers} trl peft accelerate bitsandbytes triton"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 5,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "p8GTqJOW88fI",
- "outputId": "1c7f76ee-a9ff-418d-8067-4cb357f938e5"
- },
- "outputs": [
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Valid Comet API Key saved in /root/.comet.config (set COMET_CONFIG to change where it is saved).\n",
- "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n",
- "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m As you are running in a Jupyter environment, you will need to call `experiment.end()` when finished to ensure all metrics and code are logged before exiting.\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Couldn't find a Git repository in '/content' nor in any parent directory. Set `COMET_GIT_DIRECTORY` if your Git Repository is elsewhere.\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Experiment is live on comet.com https://www.comet.com/examples/comet-example-unsloth/cb280e1a2ac942cbab2343349d80282f\n",
- "\n"
- ]
- }
- ],
- "source": [
- "import comet_ml\n",
- "\n",
- "comet_ml.login()\n",
- "exp = comet_ml.Experiment(project_name=\"comet-example-unsloth\", workspace=\"examples\")"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": null,
- "metadata": {},
- "outputs": [],
- "source": [
- "from huggingface_hub import notebook_login\n",
- "\n",
- "notebook_login()"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "5bfxeJqKwpKj"
- },
- "source": [
- "## ⚙ Download model"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 7,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "bZei5Dvw7_kM",
- "outputId": "62f7a016-6c2b-4f76-ebf2-f2051b907932"
- },
- "outputs": [
- {
- "name": "stdout",
- "output_type": "stream",
- "text": [
- "🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.\n"
- ]
- }
- ],
- "source": [
- "from unsloth import FastLanguageModel\n",
- "import torch\n",
- "\n",
- "max_seq_length = 2048\n",
- "dtype = None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+\n",
- "load_in_4bit = True"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "NG--Sbr1n5Cv"
- },
- "source": [
- "Find the full list of [100+ supported unsloth models here](https://huggingface.co/unsloth). For a full list of supported 4-bit models see [here](https://huggingface.co/collections/unsloth/load-4bit-models-4x-faster-659042e3a41c3cbad582e734)"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 8,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 284,
- "referenced_widgets": [
- "77caf8ce2875463f86cb7e5b0e02d10f",
- "986a13810ebf4e11ac2d636013efc948",
- "0902e41471fb40f7a7fb807397d1e780",
- "ec2615ff447e4a72a359143122a58887",
- "bb0052241a574fdbb6ca8c942d99e1eb",
- "1aff4958133640838a260a881e84bd66",
- "b920a8e467d64b629189d1c76541cd17",
- "0fe442a6a70948a4a9b703352148d780",
- "5bf1ce41bf6c4b389bac20669bd64e3b",
- "7d06574680cb41b9bfac0c3e47916d6e",
- "b6cd239b54c345feb11143f2364fcfaf",
- "e38214ae388e4fdca976e2adeb30d20c",
- "85d9ede0e7064521bfe7a92d54df211f",
- "740857a56a134b2dba327dc8bdbc974a",
- "725534abda3c462188fa27a978bfb59d",
- "c61150dc57cb4279b38064e1bd82b3bc",
- "6472e21298f44e6e9c5a3eefcedc59fc",
- "adb028ffb1c045f9ade7be249a33b375",
- "a1a32114bd2d4cb8882afebf56d64d41",
- "8c6ee78067c0490292ee7845dd886a1b",
- "6d28c2bb843a48908827b362ad094183",
- "41d5a34977dd4488ba5b88c308301fe6",
- "d0ce911bc4c845e88168f32781cf7923",
- "718199222cfc412e9e1ee351b45c9db6",
- "dcde9dfd56de417fb2057b3ccc9725f1",
- "c749164ce965417db5411d00f9401623",
- "61a290abfcb14c4f8b6cd9e453600bc5",
- "7cce70a2af1a4fc2a04658264c37b0c7",
- "096fb8d33b804a02b12bc085fa3c372f",
- "c6345c375f474aaf88b7b73e662b20f2",
- "b68053e4411747b0b12ab63d309aa765",
- "76664f2e75744dbf875de8fd4fdbe2d4",
- "05891b9099734df397dc8bb7396ef3e2",
- "01f7a6c4eec04fc9b063add3719c0777",
- "e4f2492f92a74794aec11bf7e2e8d554",
- "cba30ecbcad14542b3acc82e4396275c",
- "249258f9a1764d40a440a5846227059b",
- "53c336629bca4e6cacfe69ffa316890a",
- "b7c0946c9f1b40d5b469e28149808565",
- "1bc726cc52084737a0269d01f9df6a16",
- "51b3596a668742acab22e5a6659d2c55",
- "30d75e70fb6e4c4db53137804d55295d",
- "2c6dd56cf2f740cca814d5c6fa504d93",
- "55e22c9351de46c4a192008dde020393",
- "c333e3f23b8f4df6a7fd9f7a000a54ab",
- "903f318a80d849098b11451a8c855d50",
- "6b7e50badd5b4c7b979d20d7ffd1b132",
- "d83d854a33ec4adabc9bef9c2b699a0b",
- "1be75bb752214bdc8a3ef6e6976f3312",
- "d03ff62190934187be5be033a6928545",
- "1ec6fa8d08e642059f1d673f6260169f",
- "23351e3d7e8741a1adad4c1292dab841",
- "33c4f5595d924cd3a6d8c1575815a5c8",
- "6c4cf50fd9ee484396e8272591d21ee9",
- "124e70720a374644a6d360926059911f"
- ]
- },
- "id": "L09m09Xs8Y2e",
- "outputId": "4c67ce27-09cd-44dd-e987-7168a29ee335"
- },
- "outputs": [
- {
- "name": "stdout",
- "output_type": "stream",
- "text": [
- "==((====))== Unsloth 2024.8: Fast Llama patching. Transformers = 4.44.2.\n",
- " \\\\ /| GPU: NVIDIA A100-SXM4-40GB. Max memory: 39.564 GB. Platform = Linux.\n",
- "O^O/ \\_/ \\ Pytorch: 2.4.0+cu121. CUDA = 8.0. CUDA Toolkit = 12.1.\n",
- "\\ / Bfloat16 = TRUE. FA [Xformers = 0.0.27.post2. FA2 = False]\n",
- " \"-____-\" Free Apache license: http://github.com/unslothai/unsloth\n",
- "Unsloth: Fast downloading is enabled - ignore downloading bars which are red colored!\n"
- ]
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "77caf8ce2875463f86cb7e5b0e02d10f",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "model.safetensors: 0%| | 0.00/5.70G [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "e38214ae388e4fdca976e2adeb30d20c",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "generation_config.json: 0%| | 0.00/230 [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "d0ce911bc4c845e88168f32781cf7923",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "tokenizer_config.json: 0%| | 0.00/50.6k [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "01f7a6c4eec04fc9b063add3719c0777",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "tokenizer.json: 0%| | 0.00/9.09M [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "c333e3f23b8f4df6a7fd9f7a000a54ab",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "special_tokens_map.json: 0%| | 0.00/345 [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- }
- ],
- "source": [
- "model, tokenizer = FastLanguageModel.from_pretrained(\n",
- " model_name = \"unsloth/Meta-Llama-3.1-8B\",\n",
- " max_seq_length = max_seq_length,\n",
- " dtype = dtype,\n",
- " load_in_4bit = load_in_4bit,\n",
- " token = HF_TOKEN\n",
- ")"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "tbcH3PNSwRvy"
- },
- "source": [
- "## ⚙ Add LoRA adapters"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 9,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "BnXS0Trn8md9",
- "outputId": "88053e64-e9c4-4a07-a67e-c61f5647a6cc"
- },
- "outputs": [
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "Unsloth 2024.8 patched 32 layers with 32 QKV layers, 32 O layers and 32 MLP layers.\n"
- ]
- }
- ],
- "source": [
- "model = FastLanguageModel.get_peft_model(\n",
- " model,\n",
- " r = 16, # Suggested 8, 16, 32, 64, 128\n",
- " target_modules = [\"q_proj\", \"k_proj\", \"v_proj\", \"o_proj\",\n",
- " \"gate_proj\", \"up_proj\", \"down_proj\",],\n",
- " lora_alpha = 16,\n",
- " lora_dropout = 0, # Supports any, but = 0 is optimized\n",
- " bias = \"none\", # Supports any, but = \"none\" is optimized\n",
- " use_gradient_checkpointing = \"unsloth\", # True or \"unsloth\" for very long context\n",
- " random_state = 3407,\n",
- " use_rslora = False, # rank stabilized LoRA\n",
- " loftq_config = None, # LoftQ\n",
- ")"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "AoqQQ5n4wcfx"
- },
- "source": [
- "## ⚙ Data preparation"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 10,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 145,
- "referenced_widgets": [
- "ab91405c124b4e408a8c9c2ffa694b2d",
- "f92d30153e6c45b9a96ef197d5ece260",
- "3ec8fbee51cf4b72a012cd02983949af",
- "3f40bad05ed54d87a09eedce846206ed",
- "5c4d83ae4b364cd9aacc1fa045701c82",
- "a7798e04e7594d92823d43f231c5111f",
- "9af66428bb2a425bb8f941c3ec9b55c2",
- "0d35f1bb1aba4da984da6ad329509f37",
- "d1d0801d133543b599a2f11a797b8790",
- "6863f375ef34480fa548b0cf5af2f8c7",
- "6e9e00717fb642b19d38722068990c73",
- "cd56ecc4022a4d35a30213398727bd64",
- "fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
- "36d90687dc6a48d89768a3e7189e3682",
- "8697510e2823440aa123010521e81514",
- "166dde4c3cd34bbfb142b1f9b81a0c66",
- "83c3e9399bf647a9b4ac850f5c4ff158",
- "bfcefbe4f0784408a20f83a36bb908ec",
- "34e7842721d2471886a65d6038137b1c",
- "1fc06548939b4fbbaeecb7c624bfc337",
- "d6fe73f0d574413a82b08c1a59d15215",
- "39dd3663bd5d45deb2cf0b1f880f779f",
- "240d39dca5e84fb5887ec67d5ff0090f",
- "0ce31aa585754163b42c6d42f879c8d0",
- "37975894b2a349d680b50199efc7bc0f",
- "e12ed5cc35634119b1b6ee7a897fbe24",
- "6bfd07f4b40b44afbd8edff387ce6d6a",
- "77777092be6745c693d3c04d81b4dac0",
- "c86e67b875014a52b61b9bf6e31d1bf8",
- "a4acc044177d4893a7acb5d4907958ed",
- "fe1ba7f14bee4303bc5049d761e53d84",
- "f183b5acd5ea4a718841472ec865d7ac",
- "d541d6e587cb4231bd620426524479d6",
- "1809780cf0bf4855bd66beb0e67d3581",
- "9ef418a30d2643e6b4822f6c877a9b88",
- "5e0458baf7c74f43b93e7fdeba1805e3",
- "f50eebbd811e4f1a859bcdef599f6573",
- "54c7c267468d461790bb2678fe99dbee",
- "ca9dc9e749b84defab5ef912ceb0df47",
- "26dfc0e6e9a64e62b1021492ae481658",
- "69e0d359a8544795b122bfe9ab3c9b03",
- "ab1610792a52438c8a19d0ae88380c40",
- "5460a7099f1c466ea6d2feb2d69dd7ef",
- "652e3281bb4c40ebbf016d53ad680a51"
- ]
- },
- "id": "z9XFfoIz8sTI",
- "outputId": "27ec45bf-dd8c-4282-d156-e71acb2b7292"
- },
- "outputs": [
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "ab91405c124b4e408a8c9c2ffa694b2d",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Downloading readme: 0%| | 0.00/11.6k [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "cd56ecc4022a4d35a30213398727bd64",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Downloading data: 0%| | 0.00/44.3M [00:00, ?B/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "240d39dca5e84fb5887ec67d5ff0090f",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Generating train split: 0%| | 0/51760 [00:00, ? examples/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "1809780cf0bf4855bd66beb0e67d3581",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Map: 0%| | 0/51760 [00:00, ? examples/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- }
- ],
- "source": [
- "alpaca_prompt = \"\"\"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n",
- "\n",
- "### Instruction:\n",
- "{}\n",
- "\n",
- "### Input:\n",
- "{}\n",
- "\n",
- "### Response:\n",
- "{}\"\"\"\n",
- "\n",
- "EOS_TOKEN = tokenizer.eos_token # add EOS_TOKEN\n",
- "def formatting_prompts_func(examples):\n",
- " instructions = examples[\"instruction\"]\n",
- " inputs = examples[\"input\"]\n",
- " outputs = examples[\"output\"]\n",
- " texts = []\n",
- " for instruction, input, output in zip(instructions, inputs, outputs):\n",
- " # add EOS_TOKEN, otherwise your generation will go on forever\n",
- " text = alpaca_prompt.format(instruction, input, output) + EOS_TOKEN\n",
- " texts.append(text)\n",
- " return { \"text\" : texts, }\n",
- "pass\n",
- "\n",
- "from datasets import load_dataset\n",
- "dataset = load_dataset(\"yahma/alpaca-cleaned\", split = \"train\")\n",
- "dataset = dataset.map(formatting_prompts_func, batched = True,)"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "wljLKG7LwiHE"
- },
- "source": [
- "## ⚙ Training"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 11,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 67,
- "referenced_widgets": [
- "1bfcc96f38624d9d83dda4cbd60dfcc4",
- "f5612cfdb6b148c1a15e7d496012b9a5",
- "76123b2e2fa54ec0a18b306d83b4dc18",
- "abbbcf602b354fac8802267913a1a4a1",
- "3d8baf731e0f451980878320adfb5537",
- "6380d028f75d4508b71fad6deac85a41",
- "104d453482324cbebc57f44f73173fca",
- "34949a7431564226a4fc3f0f0edfa740",
- "542cef32d1874a5f9a830217a123057a",
- "cd573225788b421e92fdda6d9cb5d5d1",
- "962212c7cd9c4f6b8dad4b503780a1a9"
- ]
- },
- "id": "Ra62g_5f8vYo",
- "outputId": "bfb55505-0a0e-4aa3-9b49-f995513b33a2"
- },
- "outputs": [
- {
- "data": {
- "application/vnd.jupyter.widget-view+json": {
- "model_id": "1bfcc96f38624d9d83dda4cbd60dfcc4",
- "version_major": 2,
- "version_minor": 0
- },
- "text/plain": [
- "Map (num_proc=2): 0%| | 0/51760 [00:00, ? examples/s]"
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- },
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "max_steps is given, it will override any value given in num_train_epochs\n"
- ]
- }
- ],
- "source": [
- "from trl import SFTTrainer\n",
- "from transformers import TrainingArguments\n",
- "from unsloth import is_bfloat16_supported\n",
- "\n",
- "trainer = SFTTrainer(\n",
- " model = model,\n",
- " tokenizer = tokenizer,\n",
- " train_dataset = dataset,\n",
- " dataset_text_field = \"text\",\n",
- " max_seq_length = max_seq_length,\n",
- " dataset_num_proc = 2,\n",
- " packing = False, # Can make training 5x faster for short sequences.\n",
- " args = TrainingArguments(\n",
- " per_device_train_batch_size = 2,\n",
- " gradient_accumulation_steps = 4,\n",
- " warmup_steps = 5,\n",
- " # num_train_epochs = 1, # Set this for 1 full training run.\n",
- " max_steps = 60,\n",
- " learning_rate = 2e-4,\n",
- " fp16 = not is_bfloat16_supported(),\n",
- " bf16 = is_bfloat16_supported(),\n",
- " logging_steps = 1,\n",
- " optim = \"adamw_8bit\",\n",
- " weight_decay = 0.01,\n",
- " lr_scheduler_type = \"linear\",\n",
- " seed = 3407,\n",
- " output_dir = \"outputs\",\n",
- " ),\n",
- ")"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 12,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/",
- "height": 1000
- },
- "id": "e5EHhd5zANxX",
- "outputId": "d8381af1-03be-499e-955a-2008cf77081d"
- },
- "outputs": [
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "==((====))== Unsloth - 2x faster free finetuning | Num GPUs = 1\n",
- " \\\\ /| Num examples = 51,760 | Num Epochs = 1\n",
- "O^O/ \\_/ \\ Batch size per device = 2 | Gradient Accumulation steps = 4\n",
- "\\ / Total batch size = 8 | Total steps = 60\n",
- " \"-____-\" Number of trainable parameters = 41,943,040\n"
- ]
- },
- {
- "data": {
- "text/html": [
- "\n",
- " \n",
- " \n",
- "
\n",
- " [60/60 01:41, Epoch 0/1]\n",
- "
\n",
- " \n",
- " \n",
- " \n",
- " Step | \n",
- " Training Loss | \n",
- "
\n",
- " \n",
- " \n",
- " \n",
- " 1 | \n",
- " 1.818600 | \n",
- "
\n",
- " \n",
- " 2 | \n",
- " 2.305800 | \n",
- "
\n",
- " \n",
- " 3 | \n",
- " 1.703400 | \n",
- "
\n",
- " \n",
- " 4 | \n",
- " 2.014700 | \n",
- "
\n",
- " \n",
- " 5 | \n",
- " 1.735100 | \n",
- "
\n",
- " \n",
- " 6 | \n",
- " 1.670600 | \n",
- "
\n",
- " \n",
- " 7 | \n",
- " 1.259200 | \n",
- "
\n",
- " \n",
- " 8 | \n",
- " 1.308900 | \n",
- "
\n",
- " \n",
- " 9 | \n",
- " 1.165100 | \n",
- "
\n",
- " \n",
- " 10 | \n",
- " 1.225300 | \n",
- "
\n",
- " \n",
- " 11 | \n",
- " 0.952900 | \n",
- "
\n",
- " \n",
- " 12 | \n",
- " 0.975100 | \n",
- "
\n",
- " \n",
- " 13 | \n",
- " 0.927200 | \n",
- "
\n",
- " \n",
- " 14 | \n",
- " 1.053400 | \n",
- "
\n",
- " \n",
- " 15 | \n",
- " 0.893100 | \n",
- "
\n",
- " \n",
- " 16 | \n",
- " 0.907500 | \n",
- "
\n",
- " \n",
- " 17 | \n",
- " 1.010400 | \n",
- "
\n",
- " \n",
- " 18 | \n",
- " 1.262300 | \n",
- "
\n",
- " \n",
- " 19 | \n",
- " 1.024500 | \n",
- "
\n",
- " \n",
- " 20 | \n",
- " 0.884400 | \n",
- "
\n",
- " \n",
- " 21 | \n",
- " 0.945600 | \n",
- "
\n",
- " \n",
- " 22 | \n",
- " 1.019000 | \n",
- "
\n",
- " \n",
- " 23 | \n",
- " 0.899700 | \n",
- "
\n",
- " \n",
- " 24 | \n",
- " 0.994800 | \n",
- "
\n",
- " \n",
- " 25 | \n",
- " 1.078800 | \n",
- "
\n",
- " \n",
- " 26 | \n",
- " 1.021300 | \n",
- "
\n",
- " \n",
- " 27 | \n",
- " 1.048400 | \n",
- "
\n",
- " \n",
- " 28 | \n",
- " 0.883900 | \n",
- "
\n",
- " \n",
- " 29 | \n",
- " 0.849500 | \n",
- "
\n",
- " \n",
- " 30 | \n",
- " 0.894900 | \n",
- "
\n",
- " \n",
- " 31 | \n",
- " 0.857800 | \n",
- "
\n",
- " \n",
- " 32 | \n",
- " 0.866900 | \n",
- "
\n",
- " \n",
- " 33 | \n",
- " 0.987500 | \n",
- "
\n",
- " \n",
- " 34 | \n",
- " 0.860600 | \n",
- "
\n",
- " \n",
- " 35 | \n",
- " 0.959300 | \n",
- "
\n",
- " \n",
- " 36 | \n",
- " 0.860900 | \n",
- "
\n",
- " \n",
- " 37 | \n",
- " 0.882400 | \n",
- "
\n",
- " \n",
- " 38 | \n",
- " 0.756700 | \n",
- "
\n",
- " \n",
- " 39 | \n",
- " 1.090100 | \n",
- "
\n",
- " \n",
- " 40 | \n",
- " 1.174900 | \n",
- "
\n",
- " \n",
- " 41 | \n",
- " 0.893200 | \n",
- "
\n",
- " \n",
- " 42 | \n",
- " 0.981300 | \n",
- "
\n",
- " \n",
- " 43 | \n",
- " 0.954800 | \n",
- "
\n",
- " \n",
- " 44 | \n",
- " 0.911400 | \n",
- "
\n",
- " \n",
- " 45 | \n",
- " 0.918000 | \n",
- "
\n",
- " \n",
- " 46 | \n",
- " 0.973800 | \n",
- "
\n",
- " \n",
- " 47 | \n",
- " 0.872000 | \n",
- "
\n",
- " \n",
- " 48 | \n",
- " 1.198100 | \n",
- "
\n",
- " \n",
- " 49 | \n",
- " 0.909500 | \n",
- "
\n",
- " \n",
- " 50 | \n",
- " 1.032100 | \n",
- "
\n",
- " \n",
- " 51 | \n",
- " 1.017500 | \n",
- "
\n",
- " \n",
- " 52 | \n",
- " 0.909200 | \n",
- "
\n",
- " \n",
- " 53 | \n",
- " 0.976500 | \n",
- "
\n",
- " \n",
- " 54 | \n",
- " 1.154300 | \n",
- "
\n",
- " \n",
- " 55 | \n",
- " 0.781000 | \n",
- "
\n",
- " \n",
- " 56 | \n",
- " 1.012800 | \n",
- "
\n",
- " \n",
- " 57 | \n",
- " 0.885200 | \n",
- "
\n",
- " \n",
- " 58 | \n",
- " 0.828500 | \n",
- "
\n",
- " \n",
- " 59 | \n",
- " 0.853500 | \n",
- "
\n",
- " \n",
- " 60 | \n",
- " 0.898100 | \n",
- "
\n",
- " \n",
- "
"
- ],
- "text/plain": [
- ""
- ]
- },
- "metadata": {},
- "output_type": "display_data"
- }
- ],
- "source": [
- "trainer_stats = trainer.train()"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 13,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "8II72I92Bm9s",
- "outputId": "6518c199-dff2-4c4a-f821-aefdf595fa2c"
- },
- "outputs": [
- {
- "name": "stderr",
- "output_type": "stream",
- "text": [
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m ---------------------------------------------------------------------------------------\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Comet.ml Experiment Summary\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m ---------------------------------------------------------------------------------------\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Data:\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m display_summary_level : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m name : apparent_pagoda_1033\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m url : https://www.comet.com/examples/comet-example-unsloth/cb280e1a2ac942cbab2343349d80282f\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Metrics [count] (min, max):\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m epoch [61] : (0.0001545595054095827, 0.00927357032457496)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m grad_norm [60] : (0.2999851107597351, 2.063112258911133)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m learning_rate [60] : (0.0, 0.0002)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m loss [60] : (0.7567, 2.3058)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m total_flos : 5726714157219840.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/epoch [61] : (0.0001545595054095827, 0.00927357032457496)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/grad_norm [60] : (0.2999851107597351, 2.063112258911133)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/learning_rate [60] : (0.0, 0.0002)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/loss [60] : (0.7567, 2.3058)\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/total_flos : 5726714157219840.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_loss : 1.0665242771307628\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_runtime : 109.8838\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_samples_per_second : 4.368\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train/train_steps_per_second : 0.546\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_loss : 1.0665242771307628\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_runtime : 109.8838\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_samples_per_second : 4.368\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m train_steps_per_second : 0.546\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Others:\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m hasNestedParams : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Parameters:\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|dispatch_batches : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|even_batches : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|gradient_accumulation_kwargs : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|non_blocking : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|split_batches : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|accelerator_config|use_seedable_sampler : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adafactor : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_beta1 : 0.9\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_beta2 : 0.999\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|adam_epsilon : 1e-08\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|auto_find_batch_size : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|batch_eval_metrics : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|bf16 : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|bf16_full_eval : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|chars_per_token : \n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|data_seed : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_drop_last : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_num_workers : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_persistent_workers : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_pin_memory : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataloader_prefetch_factor : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_batch_size : 1000\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_kwargs : {}\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_num_proc : 2\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dataset_text_field : text\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_backend : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_broadcast_buffers : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_bucket_cap_mb : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_find_unused_parameters : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ddp_timeout : 1800\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|debug : []\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|deepspeed : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|disable_tqdm : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|dispatch_batches : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_eval : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_predict : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|do_train : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_accumulation_steps : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_delay : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_do_concat_batches : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_on_start : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_packing : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_steps : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_strategy : no\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|eval_use_gather_object : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|evaluation_strategy : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16 : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_backend : auto\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_full_eval : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fp16_opt_level : O1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp : []\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|min_num_params : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla_fsdp_grad_ckpt : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_config|xla_fsdp_v2 : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_min_num_params : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|fsdp_transformer_layer_cls_to_wrap : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|full_determinism : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_accumulation_steps : 4\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_checkpointing : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|gradient_checkpointing_kwargs : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|greater_is_better : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|group_by_length : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|half_precision_backend : auto\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_always_push : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_model_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_private_repo : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_strategy : every_save\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|hub_token : \n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ignore_data_skip : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_inputs_for_metrics : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_num_input_tokens_seen : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|include_tokens_per_second : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|jit_mode_eval : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|label_names : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|label_smoothing_factor : 0.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|learning_rate : 0.0002\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|length_column_name : length\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|load_best_model_at_end : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|local_rank : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_level : passive\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_level_replica : warning\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|log_on_each_node : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_dir : outputs/runs/Aug30_23-55-50_eba2d1e0a1d9\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_first_step : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_nan_inf_filter : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_steps : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|logging_strategy : steps\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|lr_scheduler_kwargs : {}\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|lr_scheduler_type : linear\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_grad_norm : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_seq_length : 2048\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|max_steps : 60\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|metric_for_best_model : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|model_init_kwargs : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|mp_parameters : \n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|neftune_noise_alpha : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|no_cuda : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|num_of_sequences : 1024\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|num_train_epochs : 3.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim : adamw_8bit\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim_args : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|optim_target_modules : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|output_dir : outputs\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|overwrite_output_dir : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|packing : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|past_index : -1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_device_eval_batch_size : 8\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_device_train_batch_size : 2\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_gpu_eval_batch_size : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|per_gpu_train_batch_size : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|prediction_loss_only : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_model_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_organization : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|push_to_hub_token : \n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|ray_scope : last\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|remove_unused_columns : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|report_to : ['comet_ml', 'tensorboard']\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|restore_callback_states_from_checkpoint : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|resume_from_checkpoint : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|run_name : outputs\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_on_each_node : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_only_model : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_safetensors : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_steps : 500\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_strategy : steps\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|save_total_limit : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|seed : 3407\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|skip_memory_metrics : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|split_batches : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tf32 : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile_backend : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_compile_mode : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torch_empty_cache_steps : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|torchdynamo : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tpu_metrics_debug : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|tpu_num_cores : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_cpu : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_ipex : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_legacy_prediction_loop : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_liger : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|use_mps_device : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|warmup_ratio : 0.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|warmup_steps : 5\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m args|weight_decay : 0.01\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|_name_or_path : unsloth/meta-llama-3.1-8b-bnb-4bit\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|add_cross_attention : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|architectures : ['LlamaForCausalLM']\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|attention_bias : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|attention_dropout : 0.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|bad_words_ids : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|begin_suppress_tokens : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|bos_token_id : 128000\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|chunk_size_feed_forward : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|cross_attention_hidden_size : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|decoder_start_token_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|diversity_penalty : 0.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|do_sample : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|early_stopping : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|encoder_no_repeat_ngram_size : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|eos_token_id : 128001\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|exponential_decay_length_penalty : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|finetuning_task : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|forced_bos_token_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|forced_eos_token_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|hidden_act : silu\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|hidden_size : 4096\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|id2label|0 : LABEL_0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|id2label|1 : LABEL_1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|initializer_range : 0.02\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|intermediate_size : 14336\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|is_decoder : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|is_encoder_decoder : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|label2id|LABEL_0 : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|label2id|LABEL_1 : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|length_penalty : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|max_length : 20\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|max_position_embeddings : 131072\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|min_length : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|mlp_bias : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|model_type : llama\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|no_repeat_ngram_size : 0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_attention_heads : 32\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_beam_groups : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_beams : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_hidden_layers : 32\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_key_value_heads : 8\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|num_return_sequences : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_attentions : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_hidden_states : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|output_scores : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pad_token_id : 128004\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|prefix : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pretraining_tp : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|problem_type : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|pruned_heads : {}\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_compute_dtype : bfloat16\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_quant_type : nf4\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|bnb_4bit_use_double_quant : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_enable_fp32_cpu_offload : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_has_fp16_weight : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_skip_modules : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|llm_int8_threshold : 6.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|load_in_4bit : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|load_in_8bit : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|quantization_config|quant_method : bitsandbytes\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|remove_invalid_values : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|repetition_penalty : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|return_dict : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|return_dict_in_generate : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rms_norm_eps : 1e-05\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|factor : 8.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|high_freq_factor : 4.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|low_freq_factor : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|original_max_position_embeddings : 8192\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_scaling|rope_type : llama3\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|rope_theta : 500000.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|sep_token_id : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|suppress_tokens : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|task_specific_params : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|temperature : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tf_legacy_loss : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tie_encoder_decoder : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tie_word_embeddings : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|tokenizer_class : None\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|top_k : 50\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|top_p : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|torch_dtype : bfloat16\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|torchscript : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|transformers_version : 4.44.2\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|typical_p : 1.0\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|unsloth_version : 2024.8\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|use_bfloat16 : False\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|use_cache : True\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m config|vocab_size : 128256\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m peft_config|default : LoraConfig(peft_type=, auto_mapping=None, base_model_name_or_path='unsloth/meta-llama-3.1-8b-bnb-4bit', revision=None, task_type=, inference_mode=False, r=16, target_modules={'k_proj', 'gate_proj', 'v_proj', 'o_proj', 'q_proj', 'down_proj', 'up_proj'}, lora_alpha=16, lora_dropout=0, fan_in_fan_out=False, bias='none', use_rslora=False, modules_to_save=None, init_lora_weights=True, layers_to_transform=None, layers_pattern=None, rank_pattern={}, alpha_pattern={}, megatron_config=None, megatron_core='megatron.core', loftq_config={}, use_dora=False, layer_replication=None, runtime_config=LoraRuntimeConfig(ephemeral_gpu_offload=False))\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m Uploads:\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m environment details : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m filename : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m installed packages : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m model graph : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m notebook : 2\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m os packages : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m source_code : 1\n",
- "\u001b[1;38;5;39mCOMET INFO:\u001b[0m \n",
- "\u001b[1;38;5;214mCOMET WARNING:\u001b[0m To get all data logged automatically, import comet_ml before the following modules: torch.\n"
- ]
- }
- ],
- "source": [
- "exp.end()"
- ]
- },
- {
- "cell_type": "markdown",
- "metadata": {
- "id": "0IdjUa_0pzPQ"
- },
- "source": [
- "## ⚙ Inference"
- ]
- },
- {
- "cell_type": "code",
- "execution_count": 12,
- "metadata": {
- "colab": {
- "base_uri": "https://localhost:8080/"
- },
- "id": "76su-yv9Aad0",
- "outputId": "8b9c2ccd-4597-41fc-d9ef-e665272fae05"
- },
- "outputs": [
- {
- "data": {
- "text/plain": [
- "['<|begin_of_text|>Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\\n\\n### Instruction:\\nContinue the fibonnaci sequence.\\n\\n### Input:\\n1, 1, 2, 3, 5, 8\\n\\n### Response:\\n13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597, 2584, 4181, 6765, 10946, 17711, 28657, 46368, 75025']"
- ]
- },
- "execution_count": 12,
- "metadata": {},
- "output_type": "execute_result"
- }
- ],
- "source": [
- "# alpaca_prompt = Copied from above\n",
- "FastLanguageModel.for_inference(model)\n",
- "inputs = tokenizer(\n",
- "[\n",
- " alpaca_prompt.format(\n",
- " \"Continue the fibonnaci sequence.\", # instruction\n",
- " \"1, 1, 2, 3, 5, 8\", # input\n",
- " \"\", # output - leave this blank for generation\n",
- " )\n",
- "], return_tensors = \"pt\").to(\"cuda\")\n",
- "\n",
- "outputs = model.generate(**inputs, max_new_tokens = 64, use_cache = True)\n",
- "tokenizer.batch_decode(outputs)"
- ]
- }
- ],
- "metadata": {
- "accelerator": "GPU",
- "colab": {
- "gpuType": "A100",
- "machine_shape": "hm",
- "provenance": [],
- "toc_visible": true
- },
- "kernelspec": {
- "display_name": "Python 3",
- "name": "python3"
- },
- "language_info": {
- "name": "python"
- },
- "widgets": {
- "application/vnd.jupyter.widget-state+json": {
- "01f7a6c4eec04fc9b063add3719c0777": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_e4f2492f92a74794aec11bf7e2e8d554",
- "IPY_MODEL_cba30ecbcad14542b3acc82e4396275c",
- "IPY_MODEL_249258f9a1764d40a440a5846227059b"
- ],
- "layout": "IPY_MODEL_53c336629bca4e6cacfe69ffa316890a"
- }
- },
- "05891b9099734df397dc8bb7396ef3e2": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "0902e41471fb40f7a7fb807397d1e780": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "danger",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_0fe442a6a70948a4a9b703352148d780",
- "max": 5702746390,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_5bf1ce41bf6c4b389bac20669bd64e3b",
- "value": 5702745847
- }
- },
- "096fb8d33b804a02b12bc085fa3c372f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "0ce31aa585754163b42c6d42f879c8d0": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_77777092be6745c693d3c04d81b4dac0",
- "placeholder": "",
- "style": "IPY_MODEL_c86e67b875014a52b61b9bf6e31d1bf8",
- "value": "Generating train split: 100%"
- }
- },
- "0d35f1bb1aba4da984da6ad329509f37": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "0fe442a6a70948a4a9b703352148d780": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "104d453482324cbebc57f44f73173fca": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "124e70720a374644a6d360926059911f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "166dde4c3cd34bbfb142b1f9b81a0c66": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1809780cf0bf4855bd66beb0e67d3581": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_9ef418a30d2643e6b4822f6c877a9b88",
- "IPY_MODEL_5e0458baf7c74f43b93e7fdeba1805e3",
- "IPY_MODEL_f50eebbd811e4f1a859bcdef599f6573"
- ],
- "layout": "IPY_MODEL_54c7c267468d461790bb2678fe99dbee"
- }
- },
- "1aff4958133640838a260a881e84bd66": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1bc726cc52084737a0269d01f9df6a16": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "1be75bb752214bdc8a3ef6e6976f3312": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "1bfcc96f38624d9d83dda4cbd60dfcc4": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_f5612cfdb6b148c1a15e7d496012b9a5",
- "IPY_MODEL_76123b2e2fa54ec0a18b306d83b4dc18",
- "IPY_MODEL_abbbcf602b354fac8802267913a1a4a1"
- ],
- "layout": "IPY_MODEL_3d8baf731e0f451980878320adfb5537"
- }
- },
- "1ec6fa8d08e642059f1d673f6260169f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "1fc06548939b4fbbaeecb7c624bfc337": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "23351e3d7e8741a1adad4c1292dab841": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "240d39dca5e84fb5887ec67d5ff0090f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_0ce31aa585754163b42c6d42f879c8d0",
- "IPY_MODEL_37975894b2a349d680b50199efc7bc0f",
- "IPY_MODEL_e12ed5cc35634119b1b6ee7a897fbe24"
- ],
- "layout": "IPY_MODEL_6bfd07f4b40b44afbd8edff387ce6d6a"
- }
- },
- "249258f9a1764d40a440a5846227059b": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_2c6dd56cf2f740cca814d5c6fa504d93",
- "placeholder": "",
- "style": "IPY_MODEL_55e22c9351de46c4a192008dde020393",
- "value": " 9.09M/9.09M [00:01<00:00, 6.98MB/s]"
- }
- },
- "26dfc0e6e9a64e62b1021492ae481658": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "2c6dd56cf2f740cca814d5c6fa504d93": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "30d75e70fb6e4c4db53137804d55295d": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "33c4f5595d924cd3a6d8c1575815a5c8": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "34949a7431564226a4fc3f0f0edfa740": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "34e7842721d2471886a65d6038137b1c": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "36d90687dc6a48d89768a3e7189e3682": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_34e7842721d2471886a65d6038137b1c",
- "max": 44307561,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_1fc06548939b4fbbaeecb7c624bfc337",
- "value": 44307561
- }
- },
- "37975894b2a349d680b50199efc7bc0f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a4acc044177d4893a7acb5d4907958ed",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_fe1ba7f14bee4303bc5049d761e53d84",
- "value": 51760
- }
- },
- "39dd3663bd5d45deb2cf0b1f880f779f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "3d8baf731e0f451980878320adfb5537": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "3ec8fbee51cf4b72a012cd02983949af": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_0d35f1bb1aba4da984da6ad329509f37",
- "max": 11610,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_d1d0801d133543b599a2f11a797b8790",
- "value": 11610
- }
- },
- "3f40bad05ed54d87a09eedce846206ed": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6863f375ef34480fa548b0cf5af2f8c7",
- "placeholder": "",
- "style": "IPY_MODEL_6e9e00717fb642b19d38722068990c73",
- "value": " 11.6k/11.6k [00:00<00:00, 16.5kB/s]"
- }
- },
- "41d5a34977dd4488ba5b88c308301fe6": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "51b3596a668742acab22e5a6659d2c55": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "53c336629bca4e6cacfe69ffa316890a": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "542cef32d1874a5f9a830217a123057a": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "5460a7099f1c466ea6d2feb2d69dd7ef": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "54c7c267468d461790bb2678fe99dbee": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "55e22c9351de46c4a192008dde020393": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "5bf1ce41bf6c4b389bac20669bd64e3b": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "5c4d83ae4b364cd9aacc1fa045701c82": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "5e0458baf7c74f43b93e7fdeba1805e3": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_69e0d359a8544795b122bfe9ab3c9b03",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_ab1610792a52438c8a19d0ae88380c40",
- "value": 51760
- }
- },
- "61a290abfcb14c4f8b6cd9e453600bc5": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6380d028f75d4508b71fad6deac85a41": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6472e21298f44e6e9c5a3eefcedc59fc": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "652e3281bb4c40ebbf016d53ad680a51": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "6863f375ef34480fa548b0cf5af2f8c7": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "69e0d359a8544795b122bfe9ab3c9b03": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6b7e50badd5b4c7b979d20d7ffd1b132": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_23351e3d7e8741a1adad4c1292dab841",
- "max": 345,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_33c4f5595d924cd3a6d8c1575815a5c8",
- "value": 345
- }
- },
- "6bfd07f4b40b44afbd8edff387ce6d6a": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6c4cf50fd9ee484396e8272591d21ee9": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6d28c2bb843a48908827b362ad094183": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "6e9e00717fb642b19d38722068990c73": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "718199222cfc412e9e1ee351b45c9db6": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_7cce70a2af1a4fc2a04658264c37b0c7",
- "placeholder": "",
- "style": "IPY_MODEL_096fb8d33b804a02b12bc085fa3c372f",
- "value": "tokenizer_config.json: 100%"
- }
- },
- "725534abda3c462188fa27a978bfb59d": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6d28c2bb843a48908827b362ad094183",
- "placeholder": "",
- "style": "IPY_MODEL_41d5a34977dd4488ba5b88c308301fe6",
- "value": " 230/230 [00:00<00:00, 20.6kB/s]"
- }
- },
- "740857a56a134b2dba327dc8bdbc974a": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a1a32114bd2d4cb8882afebf56d64d41",
- "max": 230,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_8c6ee78067c0490292ee7845dd886a1b",
- "value": 230
- }
- },
- "76123b2e2fa54ec0a18b306d83b4dc18": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_34949a7431564226a4fc3f0f0edfa740",
- "max": 51760,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_542cef32d1874a5f9a830217a123057a",
- "value": 51760
- }
- },
- "76664f2e75744dbf875de8fd4fdbe2d4": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "77777092be6745c693d3c04d81b4dac0": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "77caf8ce2875463f86cb7e5b0e02d10f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_986a13810ebf4e11ac2d636013efc948",
- "IPY_MODEL_0902e41471fb40f7a7fb807397d1e780",
- "IPY_MODEL_ec2615ff447e4a72a359143122a58887"
- ],
- "layout": "IPY_MODEL_bb0052241a574fdbb6ca8c942d99e1eb"
- }
- },
- "7cce70a2af1a4fc2a04658264c37b0c7": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "7d06574680cb41b9bfac0c3e47916d6e": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "83c3e9399bf647a9b4ac850f5c4ff158": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "85d9ede0e7064521bfe7a92d54df211f": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6472e21298f44e6e9c5a3eefcedc59fc",
- "placeholder": "",
- "style": "IPY_MODEL_adb028ffb1c045f9ade7be249a33b375",
- "value": "generation_config.json: 100%"
- }
- },
- "8697510e2823440aa123010521e81514": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_d6fe73f0d574413a82b08c1a59d15215",
- "placeholder": "",
- "style": "IPY_MODEL_39dd3663bd5d45deb2cf0b1f880f779f",
- "value": " 44.3M/44.3M [00:00<00:00, 35.4MB/s]"
- }
- },
- "8c6ee78067c0490292ee7845dd886a1b": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "903f318a80d849098b11451a8c855d50": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_d03ff62190934187be5be033a6928545",
- "placeholder": "",
- "style": "IPY_MODEL_1ec6fa8d08e642059f1d673f6260169f",
- "value": "special_tokens_map.json: 100%"
- }
- },
- "962212c7cd9c4f6b8dad4b503780a1a9": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "986a13810ebf4e11ac2d636013efc948": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_1aff4958133640838a260a881e84bd66",
- "placeholder": "",
- "style": "IPY_MODEL_b920a8e467d64b629189d1c76541cd17",
- "value": "model.safetensors: 100%"
- }
- },
- "9af66428bb2a425bb8f941c3ec9b55c2": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "9ef418a30d2643e6b4822f6c877a9b88": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_ca9dc9e749b84defab5ef912ceb0df47",
- "placeholder": "",
- "style": "IPY_MODEL_26dfc0e6e9a64e62b1021492ae481658",
- "value": "Map: 100%"
- }
- },
- "a1a32114bd2d4cb8882afebf56d64d41": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "a4acc044177d4893a7acb5d4907958ed": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "a7798e04e7594d92823d43f231c5111f": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "ab1610792a52438c8a19d0ae88380c40": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "ab91405c124b4e408a8c9c2ffa694b2d": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_f92d30153e6c45b9a96ef197d5ece260",
- "IPY_MODEL_3ec8fbee51cf4b72a012cd02983949af",
- "IPY_MODEL_3f40bad05ed54d87a09eedce846206ed"
- ],
- "layout": "IPY_MODEL_5c4d83ae4b364cd9aacc1fa045701c82"
- }
- },
- "abbbcf602b354fac8802267913a1a4a1": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_cd573225788b421e92fdda6d9cb5d5d1",
- "placeholder": "",
- "style": "IPY_MODEL_962212c7cd9c4f6b8dad4b503780a1a9",
- "value": " 51760/51760 [00:15<00:00, 3478.84 examples/s]"
- }
- },
- "adb028ffb1c045f9ade7be249a33b375": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "b68053e4411747b0b12ab63d309aa765": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "b6cd239b54c345feb11143f2364fcfaf": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "b7c0946c9f1b40d5b469e28149808565": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "b920a8e467d64b629189d1c76541cd17": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "bb0052241a574fdbb6ca8c942d99e1eb": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "bfcefbe4f0784408a20f83a36bb908ec": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "c333e3f23b8f4df6a7fd9f7a000a54ab": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_903f318a80d849098b11451a8c855d50",
- "IPY_MODEL_6b7e50badd5b4c7b979d20d7ffd1b132",
- "IPY_MODEL_d83d854a33ec4adabc9bef9c2b699a0b"
- ],
- "layout": "IPY_MODEL_1be75bb752214bdc8a3ef6e6976f3312"
- }
- },
- "c61150dc57cb4279b38064e1bd82b3bc": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "c6345c375f474aaf88b7b73e662b20f2": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "c749164ce965417db5411d00f9401623": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_76664f2e75744dbf875de8fd4fdbe2d4",
- "placeholder": "",
- "style": "IPY_MODEL_05891b9099734df397dc8bb7396ef3e2",
- "value": " 50.6k/50.6k [00:00<00:00, 236kB/s]"
- }
- },
- "c86e67b875014a52b61b9bf6e31d1bf8": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "ca9dc9e749b84defab5ef912ceb0df47": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "cba30ecbcad14542b3acc82e4396275c": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_51b3596a668742acab22e5a6659d2c55",
- "max": 9085657,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_30d75e70fb6e4c4db53137804d55295d",
- "value": 9085657
- }
- },
- "cd56ecc4022a4d35a30213398727bd64": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_fb55ed1c28fe4ebe9e9b6b0a7689c3a8",
- "IPY_MODEL_36d90687dc6a48d89768a3e7189e3682",
- "IPY_MODEL_8697510e2823440aa123010521e81514"
- ],
- "layout": "IPY_MODEL_166dde4c3cd34bbfb142b1f9b81a0c66"
- }
- },
- "cd573225788b421e92fdda6d9cb5d5d1": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d03ff62190934187be5be033a6928545": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d0ce911bc4c845e88168f32781cf7923": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_718199222cfc412e9e1ee351b45c9db6",
- "IPY_MODEL_dcde9dfd56de417fb2057b3ccc9725f1",
- "IPY_MODEL_c749164ce965417db5411d00f9401623"
- ],
- "layout": "IPY_MODEL_61a290abfcb14c4f8b6cd9e453600bc5"
- }
- },
- "d1d0801d133543b599a2f11a797b8790": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- },
- "d541d6e587cb4231bd620426524479d6": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "DescriptionStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "DescriptionStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "description_width": ""
- }
- },
- "d6fe73f0d574413a82b08c1a59d15215": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "d83d854a33ec4adabc9bef9c2b699a0b": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6c4cf50fd9ee484396e8272591d21ee9",
- "placeholder": "",
- "style": "IPY_MODEL_124e70720a374644a6d360926059911f",
- "value": " 345/345 [00:00<00:00, 26.3kB/s]"
- }
- },
- "dcde9dfd56de417fb2057b3ccc9725f1": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "FloatProgressModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "FloatProgressModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "ProgressView",
- "bar_style": "success",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_c6345c375f474aaf88b7b73e662b20f2",
- "max": 50570,
- "min": 0,
- "orientation": "horizontal",
- "style": "IPY_MODEL_b68053e4411747b0b12ab63d309aa765",
- "value": 50570
- }
- },
- "e12ed5cc35634119b1b6ee7a897fbe24": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_f183b5acd5ea4a718841472ec865d7ac",
- "placeholder": "",
- "style": "IPY_MODEL_d541d6e587cb4231bd620426524479d6",
- "value": " 51760/51760 [00:00<00:00, 136918.86 examples/s]"
- }
- },
- "e38214ae388e4fdca976e2adeb30d20c": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HBoxModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HBoxModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HBoxView",
- "box_style": "",
- "children": [
- "IPY_MODEL_85d9ede0e7064521bfe7a92d54df211f",
- "IPY_MODEL_740857a56a134b2dba327dc8bdbc974a",
- "IPY_MODEL_725534abda3c462188fa27a978bfb59d"
- ],
- "layout": "IPY_MODEL_c61150dc57cb4279b38064e1bd82b3bc"
- }
- },
- "e4f2492f92a74794aec11bf7e2e8d554": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_b7c0946c9f1b40d5b469e28149808565",
- "placeholder": "",
- "style": "IPY_MODEL_1bc726cc52084737a0269d01f9df6a16",
- "value": "tokenizer.json: 100%"
- }
- },
- "ec2615ff447e4a72a359143122a58887": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_7d06574680cb41b9bfac0c3e47916d6e",
- "placeholder": "",
- "style": "IPY_MODEL_b6cd239b54c345feb11143f2364fcfaf",
- "value": " 5.70G/5.70G [00:13<00:00, 698MB/s]"
- }
- },
- "f183b5acd5ea4a718841472ec865d7ac": {
- "model_module": "@jupyter-widgets/base",
- "model_module_version": "1.2.0",
- "model_name": "LayoutModel",
- "state": {
- "_model_module": "@jupyter-widgets/base",
- "_model_module_version": "1.2.0",
- "_model_name": "LayoutModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "LayoutView",
- "align_content": null,
- "align_items": null,
- "align_self": null,
- "border": null,
- "bottom": null,
- "display": null,
- "flex": null,
- "flex_flow": null,
- "grid_area": null,
- "grid_auto_columns": null,
- "grid_auto_flow": null,
- "grid_auto_rows": null,
- "grid_column": null,
- "grid_gap": null,
- "grid_row": null,
- "grid_template_areas": null,
- "grid_template_columns": null,
- "grid_template_rows": null,
- "height": null,
- "justify_content": null,
- "justify_items": null,
- "left": null,
- "margin": null,
- "max_height": null,
- "max_width": null,
- "min_height": null,
- "min_width": null,
- "object_fit": null,
- "object_position": null,
- "order": null,
- "overflow": null,
- "overflow_x": null,
- "overflow_y": null,
- "padding": null,
- "right": null,
- "top": null,
- "visibility": null,
- "width": null
- }
- },
- "f50eebbd811e4f1a859bcdef599f6573": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_5460a7099f1c466ea6d2feb2d69dd7ef",
- "placeholder": "",
- "style": "IPY_MODEL_652e3281bb4c40ebbf016d53ad680a51",
- "value": " 51760/51760 [00:00<00:00, 113885.51 examples/s]"
- }
- },
- "f5612cfdb6b148c1a15e7d496012b9a5": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_6380d028f75d4508b71fad6deac85a41",
- "placeholder": "",
- "style": "IPY_MODEL_104d453482324cbebc57f44f73173fca",
- "value": "Map (num_proc=2): 100%"
- }
- },
- "f92d30153e6c45b9a96ef197d5ece260": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_a7798e04e7594d92823d43f231c5111f",
- "placeholder": "",
- "style": "IPY_MODEL_9af66428bb2a425bb8f941c3ec9b55c2",
- "value": "Downloading readme: 100%"
- }
- },
- "fb55ed1c28fe4ebe9e9b6b0a7689c3a8": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "HTMLModel",
- "state": {
- "_dom_classes": [],
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "HTMLModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/controls",
- "_view_module_version": "1.5.0",
- "_view_name": "HTMLView",
- "description": "",
- "description_tooltip": null,
- "layout": "IPY_MODEL_83c3e9399bf647a9b4ac850f5c4ff158",
- "placeholder": "",
- "style": "IPY_MODEL_bfcefbe4f0784408a20f83a36bb908ec",
- "value": "Downloading data: 100%"
- }
- },
- "fe1ba7f14bee4303bc5049d761e53d84": {
- "model_module": "@jupyter-widgets/controls",
- "model_module_version": "1.5.0",
- "model_name": "ProgressStyleModel",
- "state": {
- "_model_module": "@jupyter-widgets/controls",
- "_model_module_version": "1.5.0",
- "_model_name": "ProgressStyleModel",
- "_view_count": null,
- "_view_module": "@jupyter-widgets/base",
- "_view_module_version": "1.2.0",
- "_view_name": "StyleView",
- "bar_color": null,
- "description_width": ""
- }
- }
- }
- }
+ "cells": [
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "6RF2EQaKDoxr"
+ },
+ "source": [
+ "\n",
+ " \n",
+ ""
+ ]
},
- "nbformat": 4,
- "nbformat_minor": 0
-}
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "KDjO5WbeDtp0"
+ },
+ "source": [
+ "# Comet and unsloth\n",
+ "\n",
+ "[Comet](https://www.comet.com/site/?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) is an MLOps platform designed to help data scientists and teams build better models faster! Comet provides tooling to track, explain, manage, and monitor your models in a single place! It works with Jupyter notebooks and scripts and-- most importantly--it's 100% free to get started!\n",
+ "\n",
+ "[unsloth](https://github.com/unslothai/unsloth) dramatically improves the speed and efficiency of LLM fine-tuning for models including Llama, Phi-3, Gemma, Mistral, and more. For a full listed of 100+ supported unsloth models, [see here](https://huggingface.co/unsloth).\n",
+ "\n",
+ "Instrument your torchtune training runs with Comet to start managing experiments with efficiency, reproducibility, and collaboration in mind.\n",
+ "\n",
+ "Find more information about [our integration with torchtune here](https://www.comet.com/docs/v2/integrations/third-party-tools/unsloth?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth) or [learn about our other integrations here](https://www.comet.com/docs/v2/integrations?utm_medium=colab&utm_source=comet-examples&utm_campaign=unsloth)."
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "f0fhpJZgsYII"
+ },
+ "source": [
+ "## ⚙ Install and import dependencies"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "5zTSjY9r4cGc"
+ },
+ "outputs": [],
+ "source": [
+ "%pip install comet_ml \"unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git\" \"torch>=2.4.0\" xformers trl peft accelerate bitsandbytes triton"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "p8GTqJOW88fI"
+ },
+ "outputs": [],
+ "source": [
+ "import comet_ml\n",
+ "\n",
+ "comet_ml.login()\n",
+ "exp = comet_ml.Experiment(project_name=\"comet-example-unsloth\")"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "vdOozTnqq8pL"
+ },
+ "outputs": [],
+ "source": [
+ "from huggingface_hub import notebook_login\n",
+ "\n",
+ "notebook_login()"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "5bfxeJqKwpKj"
+ },
+ "source": [
+ "## ⚙ Download model"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "bZei5Dvw7_kM"
+ },
+ "outputs": [],
+ "source": [
+ "from unsloth import FastLanguageModel\n",
+ "import torch\n",
+ "\n",
+ "max_seq_length = 2048\n",
+ "dtype = (\n",
+ " None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+\n",
+ ")\n",
+ "load_in_4bit = True"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "NG--Sbr1n5Cv"
+ },
+ "source": [
+ "Find the full list of [100+ supported unsloth models here](https://huggingface.co/unsloth). For a full list of supported 4-bit models see [here](https://huggingface.co/collections/unsloth/load-4bit-models-4x-faster-659042e3a41c3cbad582e734)"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "L09m09Xs8Y2e"
+ },
+ "outputs": [],
+ "source": [
+ "model, tokenizer = FastLanguageModel.from_pretrained(\n",
+ " model_name=\"unsloth/Meta-Llama-3.1-8B\",\n",
+ " max_seq_length=max_seq_length,\n",
+ " dtype=dtype,\n",
+ " load_in_4bit=load_in_4bit,\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "tbcH3PNSwRvy"
+ },
+ "source": [
+ "## ⚙ Add LoRA adapters"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "BnXS0Trn8md9"
+ },
+ "outputs": [],
+ "source": [
+ "model = FastLanguageModel.get_peft_model(\n",
+ " model,\n",
+ " r=16, # Suggested 8, 16, 32, 64, 128\n",
+ " target_modules=[\n",
+ " \"q_proj\",\n",
+ " \"k_proj\",\n",
+ " \"v_proj\",\n",
+ " \"o_proj\",\n",
+ " \"gate_proj\",\n",
+ " \"up_proj\",\n",
+ " \"down_proj\",\n",
+ " ],\n",
+ " lora_alpha=16,\n",
+ " lora_dropout=0, # Supports any, but = 0 is optimized\n",
+ " bias=\"none\", # Supports any, but = \"none\" is optimized\n",
+ " use_gradient_checkpointing=\"unsloth\", # True or \"unsloth\" for very long context\n",
+ " random_state=3407,\n",
+ " use_rslora=False, # rank stabilized LoRA\n",
+ " loftq_config=None, # LoftQ\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "AoqQQ5n4wcfx"
+ },
+ "source": [
+ "## ⚙ Data preparation"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "z9XFfoIz8sTI"
+ },
+ "outputs": [],
+ "source": [
+ "alpaca_prompt = \"\"\"Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.\n",
+ "\n",
+ "### Instruction:\n",
+ "{}\n",
+ "\n",
+ "### Input:\n",
+ "{}\n",
+ "\n",
+ "### Response:\n",
+ "{}\"\"\"\n",
+ "\n",
+ "EOS_TOKEN = tokenizer.eos_token # add EOS_TOKEN\n",
+ "\n",
+ "\n",
+ "def formatting_prompts_func(examples):\n",
+ " instructions = examples[\"instruction\"]\n",
+ " inputs = examples[\"input\"]\n",
+ " outputs = examples[\"output\"]\n",
+ " texts = []\n",
+ " for instruction, input, output in zip(instructions, inputs, outputs):\n",
+ " # add EOS_TOKEN, otherwise your generation will go on forever\n",
+ " text = alpaca_prompt.format(instruction, input, output) + EOS_TOKEN\n",
+ " texts.append(text)\n",
+ " return {\n",
+ " \"text\": texts,\n",
+ " }\n",
+ "\n",
+ "\n",
+ "pass\n",
+ "\n",
+ "from datasets import load_dataset\n",
+ "\n",
+ "dataset = load_dataset(\"yahma/alpaca-cleaned\", split=\"train\")\n",
+ "dataset = dataset.map(\n",
+ " formatting_prompts_func,\n",
+ " batched=True,\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "wljLKG7LwiHE"
+ },
+ "source": [
+ "## ⚙ Training"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "Ra62g_5f8vYo"
+ },
+ "outputs": [],
+ "source": [
+ "from trl import SFTTrainer\n",
+ "from transformers import TrainingArguments\n",
+ "from unsloth import is_bfloat16_supported\n",
+ "\n",
+ "trainer = SFTTrainer(\n",
+ " model=model,\n",
+ " tokenizer=tokenizer,\n",
+ " train_dataset=dataset,\n",
+ " dataset_text_field=\"text\",\n",
+ " max_seq_length=max_seq_length,\n",
+ " dataset_num_proc=2,\n",
+ " packing=False, # Can make training 5x faster for short sequences.\n",
+ " args=TrainingArguments(\n",
+ " per_device_train_batch_size=2,\n",
+ " gradient_accumulation_steps=4,\n",
+ " warmup_steps=5,\n",
+ " # num_train_epochs = 1, # Set this for 1 full training run.\n",
+ " max_steps=60,\n",
+ " learning_rate=2e-4,\n",
+ " fp16=not is_bfloat16_supported(),\n",
+ " bf16=is_bfloat16_supported(),\n",
+ " logging_steps=1,\n",
+ " optim=\"adamw_8bit\",\n",
+ " weight_decay=0.01,\n",
+ " lr_scheduler_type=\"linear\",\n",
+ " seed=3407,\n",
+ " output_dir=\"outputs\",\n",
+ " ),\n",
+ ")"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "e5EHhd5zANxX"
+ },
+ "outputs": [],
+ "source": [
+ "trainer_stats = trainer.train()"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "8II72I92Bm9s"
+ },
+ "outputs": [],
+ "source": [
+ "exp.end()"
+ ]
+ },
+ {
+ "cell_type": "markdown",
+ "metadata": {
+ "id": "0IdjUa_0pzPQ"
+ },
+ "source": [
+ "## ⚙ Inference"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {
+ "id": "76su-yv9Aad0"
+ },
+ "outputs": [],
+ "source": [
+ "# alpaca_prompt = Copied from above\n",
+ "FastLanguageModel.for_inference(model)\n",
+ "inputs = tokenizer(\n",
+ " [\n",
+ " alpaca_prompt.format(\n",
+ " \"Continue the fibonnaci sequence.\", # instruction\n",
+ " \"1, 1, 2, 3, 5, 8\", # input\n",
+ " \"\", # output - leave this blank for generation\n",
+ " )\n",
+ " ],\n",
+ " return_tensors=\"pt\",\n",
+ ").to(\"cuda\")\n",
+ "\n",
+ "outputs = model.generate(**inputs, max_new_tokens=64, use_cache=True)\n",
+ "tokenizer.batch_decode(outputs)"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "source": [],
+ "metadata": {
+ "id": "h6alQJYpwlAM"
+ },
+ "execution_count": null,
+ "outputs": []
+ }
+ ],
+ "metadata": {
+ "accelerator": "GPU",
+ "colab": {
+ "gpuType": "L4",
+ "machine_shape": "hm",
+ "provenance": [],
+ "toc_visible": true
+ },
+ "kernelspec": {
+ "display_name": "Python 3",
+ "name": "python3"
+ },
+ "language_info": {
+ "name": "python"
+ }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 0
+}
\ No newline at end of file