`torch_dtype` is deprecated! Use `dtype` instead!

Loading checkpoint shards:   0%|          | 0/4 [00:00<?, ?it/s]
Loading checkpoint shards:  25%|██▌       | 1/4 [00:03<00:10,  3.54s/it]
Loading checkpoint shards:  50%|█████     | 2/4 [00:07<00:07,  3.66s/it]
Loading checkpoint shards:  75%|███████▌  | 3/4 [00:10<00:03,  3.54s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:12<00:00,  2.71s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:12<00:00,  3.03s/it]

Loading checkpoint shards:   0%|          | 0/4 [00:00<?, ?it/s]
Loading checkpoint shards:  25%|██▌       | 1/4 [00:03<00:11,  3.95s/it]
Loading checkpoint shards:  50%|█████     | 2/4 [00:07<00:07,  3.74s/it]
Loading checkpoint shards:  75%|███████▌  | 3/4 [00:11<00:03,  3.70s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:11<00:00,  2.43s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:11<00:00,  2.92s/it]

Loading checkpoint shards:   0%|          | 0/5 [00:00<?, ?it/s]
Loading checkpoint shards:  20%|██        | 1/5 [00:04<00:16,  4.24s/it]
Loading checkpoint shards:  40%|████      | 2/5 [00:07<00:11,  3.81s/it]
Loading checkpoint shards:  60%|██████    | 3/5 [00:11<00:07,  3.68s/it]
Loading checkpoint shards:  80%|████████  | 4/5 [00:13<00:03,  3.22s/it]
Loading checkpoint shards: 100%|██████████| 5/5 [00:14<00:00,  2.46s/it]
Loading checkpoint shards: 100%|██████████| 5/5 [00:14<00:00,  2.98s/it]

Loading checkpoint shards:   0%|          | 0/4 [00:00<?, ?it/s]
Loading checkpoint shards:  25%|██▌       | 1/4 [00:04<00:13,  4.49s/it]
Loading checkpoint shards:  50%|█████     | 2/4 [00:08<00:08,  4.42s/it]
Loading checkpoint shards:  75%|███████▌  | 3/4 [00:12<00:04,  4.17s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:14<00:00,  3.17s/it]
Loading checkpoint shards: 100%|██████████| 4/4 [00:14<00:00,  3.59s/it]
Traceback (most recent call last):
  File "/projects/bfqt/users/yurenh2/ml-projects/personalization-user-model/collaborativeagents/scripts/test_multiturn.py", line 226, in <module>
    results["full_session"] = test_full_session()
                              ^^^^^^^^^^^^^^^^^^^
  File "/projects/bfqt/users/yurenh2/ml-projects/personalization-user-model/collaborativeagents/scripts/test_multiturn.py", line 171, in test_full_session
    adapter = create_baseline_adapter("vanilla")
              ^^^^^^^^^^^^^^^^^^^^^^^
NameError: name 'create_baseline_adapter' is not defined