truss 0.10.9rc535__py3-none-any.whl → 0.10.10rc0__py3-none-any.whl

{truss-0.10.9rc535.dist-info → truss-0.10.10rc0.dist-info}/RECORD

@@ -11,22 +11,24 @@ truss/base/truss_spec.py,sha256=jFVF79CXoEEspl2kXBAPyi-rwISReIGTdobGpaIhwJw,5979
 truss/cli/chains_commands.py,sha256=y6pdIAGCcKOPG9bPuCXPfSA0onQm5x-tT_3blSBfPYg,16971
 truss/cli/cli.py,sha256=PaMkuwXZflkU7sa1tEoT_Zmy-iBkEZs1m4IVqcieaeo,30367
 truss/cli/remote_cli.py,sha256=G_xCKRXzgkCmkiZJhUFfsv5YSVgde1jLA5LPQitpZgI,1905
-truss/cli/train_commands.py,sha256=VPwlyfCUumRpwN1m9hkBxHKqobUmIbeQnfqnsl6_uPo,12078
-truss/cli/logs/base_watcher.py,sha256=1qu0bKLKSB7aVociYrdpJD_ItjW8C1t5P6zr1Ku87YU,2785
+truss/cli/train_commands.py,sha256=P9bdnpq1SgEGXBaVf9joKdsaCDX2v29P4MhLMuz-jYw,12344
+truss/cli/logs/base_watcher.py,sha256=KKyd7lIrdaEeDVt8EtjMioSPGVpLyOcF0ewyzE_GGdQ,2785
 truss/cli/logs/model_log_watcher.py,sha256=NACcP-wkcaroYa2Cb9BZC7Yr0554WZa_FSM2LXOf4A8,1263
 truss/cli/logs/training_log_watcher.py,sha256=r6HRqrLnz-PiKTUXiDYYxg4ZnP8vYcXlEX1YmgHhzlo,1173
 truss/cli/logs/utils.py,sha256=z-U_FG4BUzdZLbE3BnXb4DZQ0zt3LSZ3PiQpLaDuc3o,1031
 truss/cli/train/common.py,sha256=Es1yllSYxjM9x2uBzTGbYwyd8ML66cqqge0XO8_G_X0,992
 truss/cli/train/core.py,sha256=MBOhPSVYOU7wVh09uWQrJDEVOhJQug_2Odv3u6tCVTA,13855
 truss/cli/train/deploy_from_checkpoint_config.yml,sha256=mktaVrfhN8Kjx1UveC4xr-gTW-kjwbHvq6bx_LpO-Wg,371
-truss/cli/train/metrics_watcher.py,sha256=iDD06zt7ze6thy9APE-SlyHwUdetWmbKbN7NUp1jE1U,9128
+truss/cli/train/deploy_from_checkpoint_config_whisper.yml,sha256=6GbOorYC8ml0UyOUvuBpFO_fuYtYE646JqsalR-D4oY,406
+truss/cli/train/metrics_watcher.py,sha256=ftrLQ5m7V1lAqcAvdGbMv5r0aF4D0lypfKjokCBQvLw,12798
 truss/cli/train/poller.py,sha256=TGRzELxsicga0bEXewSX1ujw6lfPmDnHd6nr8zvOFO8,3550
 truss/cli/train/types.py,sha256=alGtr4Q71GeB65PpGMhsoKygw4k_ncR6MKIP1ioP8rI,951
 truss/cli/train/deploy_checkpoints/__init__.py,sha256=wL-M2yu8PxO2tFvjwshXAfPnB-5TlvsBp2v_bdzimRU,99
-truss/cli/train/deploy_checkpoints/deploy_checkpoints.py,sha256=wWFCpaMjM868-f_ChECrGHyMHka1F2VS1pEKPcEw3eM,16667
-truss/cli/train/deploy_checkpoints/deploy_checkpoints_helpers.py,sha256=7CdYgsxDF7nHITyCGpjjTMeaaOvtlCYwF7NSxpKedS0,1723
-truss/cli/train/deploy_checkpoints/deploy_full_checkpoints.py,sha256=pFj7rDcnvb9C4MMfr3wc4aBXfziqFkzw0H883NtQ1Es,3245
+truss/cli/train/deploy_checkpoints/deploy_checkpoints.py,sha256=xfblHi3py7GDgY24NcuAaDKzcQeOm67rjtWOK6vAEe4,17352
+truss/cli/train/deploy_checkpoints/deploy_checkpoints_helpers.py,sha256=6x5nS_HnWYtS9vi-Pg8akzrJk9L_agjvFhm5EFh1m6Y,1964
+truss/cli/train/deploy_checkpoints/deploy_full_checkpoints.py,sha256=FYRG5KTMlxEMZS-RA_m2gp1wuqWbSpqt2RhdQfLibhA,3968
 truss/cli/train/deploy_checkpoints/deploy_lora_checkpoints.py,sha256=P91dIAzuhl2GlzmrWwCcYI7uCMT1Lm7C79JQHM_exN4,4442
+truss/cli/train/deploy_checkpoints/deploy_whisper_checkpoints.py,sha256=NSo2kEn-CxawGvUhn-xE81noxoTJ0cCv8X9fkrMxAsM,2617
 truss/cli/utils/common.py,sha256=aWnla4qMSEz57dRMTl7R-EaScsuEpnQUeziGUaIeqeU,6149
 truss/cli/utils/output.py,sha256=GNjU85ZAMp5BI6Yij5wYXcaAvpm_kmHV0nHNmdkMxb0,646
 truss/cli/utils/self_upgrade.py,sha256=eTJZA4Wc8uUp4Qh6viRQp6bZm--wnQp7KWe5KRRpPtg,5427
@@ -34,7 +36,7 @@ truss/contexts/docker_build_setup.py,sha256=cF4ExZgtYvrWxvyCAaUZUvV_DB_7__MqVomU
 truss/contexts/truss_context.py,sha256=uS6L-ACHxNk0BsJwESOHh1lA0OGGw0pb33aFKGsASj4,436
 truss/contexts/image_builder/cache_warmer.py,sha256=TGMV1Mh87n2e_dSowH0sf0rZhZraDOR-LVapZL3a5r8,7377
 truss/contexts/image_builder/image_builder.py,sha256=IuRgDeeoHVLzIkJvKtX3807eeqEyaroCs_KWDcIHZUg,1461
-truss/contexts/image_builder/serving_image_builder.py,sha256=UnChWhK_DS6NlfucmivZr8LyI62X0b-ZiuIdFYDqBQY,34209
+truss/contexts/image_builder/serving_image_builder.py,sha256=FH5HPnrr9_OomN5WplsyUrGGETe9ld6h3q9JCpvB6FY,33322
 truss/contexts/image_builder/util.py,sha256=y2-CjUKv0XV-0w2sr1fUCflysDJLsoU4oPp6tvvoFnk,1203
 truss/contexts/local_loader/docker_build_emulator.py,sha256=rmf7I28zksSmHjwvJMx2rIa6xK4KeR5fBm5YFth_fQg,2464
 truss/contexts/local_loader/dockerfile_parser.py,sha256=GoRJ0Af_3ILyLhjovK5lrCGn1rMxz6W3l681ro17ZzI,1344
@@ -50,9 +52,9 @@ truss/patch/truss_dir_patch_applier.py,sha256=ALnaVnu96g0kF2UmGuBFTua3lrXpwAy4sG
 truss/remote/remote_factory.py,sha256=-0gLh_yIyNDgD48Q6sR8Yo5dOMQg84lrHRvn_XR0n4s,3585
 truss/remote/truss_remote.py,sha256=TEe6h6by5-JLy7PMFsDN2QxIY5FmdIYN3bKvHHl02xM,8440
 truss/remote/baseten/__init__.py,sha256=XNqJW1zyp143XQc6-7XVwsUA_Q_ZJv_ausn1_Ohtw9Y,176
-truss/remote/baseten/api.py,sha256=yVe4Ees-MXJucet2y1asNPQE4N1W1YB80tSy6VC4TBk,24074
+truss/remote/baseten/api.py,sha256=6Nie4hv4z5I62boeCQvP3tGA0Pwu96bMgz1vp5Tkxao,24447
 truss/remote/baseten/auth.py,sha256=tI7s6cI2EZgzpMIzrdbILHyGwiHDnmoKf_JBhJXT55E,776
-truss/remote/baseten/core.py,sha256=ZkiXi76lCM8F8111tzVtke-nobHGPcOxGb52PoqEr5M,14708
+truss/remote/baseten/core.py,sha256=uxtmBI9RAVHu1glIEJb5Q4ccJYLeZM1Cp5Svb9W68Yw,21965
 truss/remote/baseten/custom_types.py,sha256=g7MwgYaeqIxF-e170G5iEVLWiw5jgAnqXztIUqVkdyc,3227
 truss/remote/baseten/error.py,sha256=3TNTwwPqZnr4NRd9Sl6SfLUQR2fz9l6akDPpOntTpzA,578
 truss/remote/baseten/remote.py,sha256=Se8AES5mk8jxa8S9fN2DSG7wnsaV7ftRjJ4Uwc_w_S0,22544
@@ -60,15 +62,16 @@ truss/remote/baseten/rest_client.py,sha256=_t3CWsWARt2u0C0fDsF4rtvkkHe-lH7KXoPxW
 truss/remote/baseten/service.py,sha256=j_cCbSkpvCqIoptCDp65BCRKORPrp9NswTdI3BbiFqU,6081
 truss/remote/baseten/utils/status.py,sha256=jputc9N9AHXxUuW4KOk6mcZKzQ_gOBOe5BSx9K0DxPY,1266
 truss/remote/baseten/utils/tar.py,sha256=pMUv--YkwXDngUx1WUOK-KmAIKMcOg2E-CD5x4heh3s,2514
+truss/remote/baseten/utils/time.py,sha256=Ry9GMjYnbIGYVIGwtmv4V8ljWjvdcaCf5NOQzlNeGxI,397
 truss/remote/baseten/utils/transfer.py,sha256=d3VptuQb6M1nyS6kz0BAfeOYDLkMKUjatJXpY-mp-As,1548
 truss/templates/README.md.jinja,sha256=N7CJdyldZuJamj5jLh47le0hFBdu9irVsTBqoxhPNPQ,2476
 truss/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-truss/templates/base.Dockerfile.jinja,sha256=sSisbWuVAHJBlIgmC6xJu1032zmjiZX5Gbt00djvRjk,4859
-truss/templates/cache.Dockerfile.jinja,sha256=evMbbMXjceDjUmGGuB0shLS_PxZsc9ZGkbzR-6aIW-Q,1114
+truss/templates/base.Dockerfile.jinja,sha256=vFAJH1lC9jg90-076H2DCmkXUAlpseitIN6c4UwagxA,4020
+truss/templates/cache.Dockerfile.jinja,sha256=LhsVP9F3BATKQGkgya_YT4v6ABTUkpy-Jb3N36zsw10,1030
 truss/templates/cache_requirements.txt,sha256=xoPoJ-OVnf1z6oq_RVM3vCr3ionByyqMLj7wGs61nUs,87
-truss/templates/copy_cache_files.Dockerfile.jinja,sha256=Os5zFdYLZ_AfCRGq4RcpVTObOTwL7zvmwYcvOzd_Zqo,126
+truss/templates/copy_cache_files.Dockerfile.jinja,sha256=arHldnuclt7vUFHyRz6vus5NGMDkIofm-1RU37A0xZM,98
 truss/templates/docker_server_requirements.txt,sha256=PyhOPKAmKW1N2vLvTfLMwsEtuGpoRrbWuNo7tT6v2Mc,18
-truss/templates/server.Dockerfile.jinja,sha256=tjCYmolZHR4d0BXiJUKkTP6nGh9XHjY7ixDl9ShYrrA,7252
+truss/templates/server.Dockerfile.jinja,sha256=RDkP4_6nOw0rZUOZBkDS4leOTTBUbSb8buvx1hi8ncY,5996
 truss/templates/control/requirements.txt,sha256=Kk0tYID7trPk5gwX38Wrt2-YGWZAXFJCJRcqJ8ZzCjc,251
 truss/templates/control/control/application.py,sha256=jYeta6hWe1SkfLL3W4IDmdYjg3ZuKqI_UagWYs5RB_E,3793
 truss/templates/control/control/endpoints.py,sha256=FM-sgao7I3gMoUTasM3Xq_g2LDoJQe75JxIoaQxzeNo,10031
@@ -89,11 +92,11 @@ truss/templates/custom/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 truss/templates/custom/model/model.py,sha256=J04rLxK09Pwt2F4GoKOLKL-H-CqZUdYIM-PL2CE9PoE,1079
 truss/templates/custom_python_dx/my_model.py,sha256=NG75mQ6wxzB1BYUemDFZvRLBET-UrzuUK4FuHjqI29U,910
 truss/templates/docker_server/proxy.conf.jinja,sha256=Lg-PcZzKflG85exZKHNgW_I6r0mATV8AtOIBaE40-RM,1669
-truss/templates/docker_server/supervisord.conf.jinja,sha256=dd37fwZE--cutrvOUCqEyJQQQhlp61H2IUs2huKWsSk,1808
+truss/templates/docker_server/supervisord.conf.jinja,sha256=CoaSLv0Lr8t1tS_q102IFufNX2lWrlbCHJLjMhYjOwM,1711
 truss/templates/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 truss/templates/server/main.py,sha256=kWXrdD8z8IpamyWxc8qcvd5ck9gM1Kz2QH5qHJCnmOQ,222
 truss/templates/server/model_wrapper.py,sha256=k75VVISwwlsx5EGb82UZsu8kCM_i6Yi3-Hd0-Kpm1yo,42055
-truss/templates/server/requirements.txt,sha256=iRR2BEpBQnt-YOiTEKOnaab7tlR4C23V1cuURuIt7ZY,672
+truss/templates/server/requirements.txt,sha256=1pIQJ0US7KnMxC09TUYj09nF1Dssdg2xWTY9yilfFDQ,672
 truss/templates/server/truss_server.py,sha256=ob_nceeGtFPZzKKdk_ZZGLoZrJOGE6hR52xM1sPR97A,19498
 truss/templates/server/common/__init__.py,sha256=qHIqr68L5Tn4mV6S-PbORpcuJ4jmtBR8aCuRTIWDvNo,85
 truss/templates/server/common/errors.py,sha256=qWeZlmNI8ZGbZbOIp_mtS6IKvUFIzhj3QH8zp-xTp9o,8554
@@ -136,8 +139,8 @@ truss/tests/test_truss_gatherer.py,sha256=bn288OEkC49YY0mhly4cAl410ktZPfElNdWwZy
 truss/tests/test_truss_handle.py,sha256=-xz9VXkecXDTslmQZ-dmUmQLnvD0uumRqHS2uvGlMBA,30750
 truss/tests/test_util.py,sha256=hs1bNMkXKEdoPRx4Nw-NAEdoibR92OubZuADGmbiYsQ,1344
 truss/tests/cli/test_cli.py,sha256=yfbVS5u1hnAmmA8mJ539vj3lhH-JVGUvC4Q_Mbort44,787
-truss/tests/cli/train/test_deploy_checkpoints.py,sha256=pPTqyGqvSsrsEHXguo6swLXkVObytLdupfSElqlujPc,25733
-truss/tests/cli/train/test_train_cli_core.py,sha256=ONI4Oc1R0pL-M3WuNsnnHLzqEfMTE0u9g5dE9Ii9yW8,3424
+truss/tests/cli/train/test_deploy_checkpoints.py,sha256=wQZ3DPLPAyXE3iaQiyHJTBO15v_gXN44eDk1StYkKmM,44764
+truss/tests/cli/train/test_train_cli_core.py,sha256=T1Xa6-NRk2nTJGX6sXaA8x4qCwL3Ini72PBI2gW7rYM,7879
 truss/tests/cli/train/resources/test_deploy_from_checkpoint_config.yml,sha256=GF7r9l0KaeXiUYCPSBpeMPd2QG6PeWWyI12NdbqLOgc,1930
 truss/tests/contexts/image_builder/test_serving_image_builder.py,sha256=iJA7nxcLXhBmyjhLIKeN64ql0OI_R53l-qSt3SsENV8,22368
 truss/tests/contexts/local_loader/test_load_local.py,sha256=D1qMH2IpYA2j5009v50QMgUnKdeOsX15ndkwXe10a4E,801
@@ -151,9 +154,10 @@ truss/tests/patch/test_truss_dir_patch_applier.py,sha256=P0SCqkVXLge7laSPlFWZM7A
 truss/tests/patch/test_types.py,sha256=OUVDiLckbjjjEN49I4hm62emOTAr8lv_QooJrmXxs5o,306
 truss/tests/remote/test_remote_factory.py,sha256=S-iZlF5Pf5SDoFUnMlZXy9iRMkosVgwLd22evzWlFr0,4842
 truss/tests/remote/test_truss_remote.py,sha256=Rguyrnbx5RlbPJHFfCtsRtX1czAJ9Fo0aeC5EWRVkGw,2726
-truss/tests/remote/baseten/test_api.py,sha256=2UMeKrXSi6JtaUpa1B-mZVY13HuYvkdQsUuGigCUiHE,14668
+truss/tests/remote/baseten/conftest.py,sha256=vNk0nfDB7XdmqatOMhjdANCWFGYM4VwSHVKlaBO2PPk,442
+truss/tests/remote/baseten/test_api.py,sha256=AKJeNsrUtTNa0QPClfEvXlBOSJ214PKp23ULehMRJOQ,15885
 truss/tests/remote/baseten/test_auth.py,sha256=ttu4bDnmwGfo3oiNut4HVGnh-QnjAefwZJctiibQJKY,669
-truss/tests/remote/baseten/test_core.py,sha256=NiU_8EKLzKLIUSfr3-z0sQ7ruDBg4pcOxSoeFOq7wqo,8176
+truss/tests/remote/baseten/test_core.py,sha256=6NzJTDmoSUv6Muy1LFEYIUg10-cqw-hbLyeTSWcdNjY,26117
 truss/tests/remote/baseten/test_remote.py,sha256=y1qSPL1t7dBeYI3xMFn436fttG7wkYdAoENTz7qKObg,23634
 truss/tests/remote/baseten/test_service.py,sha256=ufZbtQlBNIzFCxRt_iE-APLpWbVw_3ViUpSh6H9W5nU,1945
 truss/tests/templates/control/control/test_endpoints.py,sha256=tGU3w8zOKC8LfWGdhp-TlV7E603KXg2xGwpqDdf8Pnw,3385
@@ -305,7 +309,6 @@ truss/tests/test_data/test_truss_with_error/packages/helpers_1.py,sha256=qIm-hQY
 truss/tests/test_data/test_truss_with_error/packages/helpers_2.py,sha256=q_UpVfXq_K2tuHv6YwsIzVHC3sy5k5hKDw6lMCdS0oc,53
 truss/tests/trt_llm/test_trt_llm_config.py,sha256=lNQ4EEkOsiT17KvnvW1snCeEBd7K_cl9_Y0dko3qpn8,8505
 truss/tests/trt_llm/test_validation.py,sha256=dmax2EHxRfqxJvWzV8uubkTef50833KBBHw-WkHufL8,2120
-truss/tests/util/test_basetenpointer.py,sha256=Bdms21_m8T4xmFNHRO5nS2tU2wU7094_1SkfBxjptmk,9824
 truss/tests/util/test_config_checks.py,sha256=aoZF_Q-eRd3qz5wjUqa8Cr_7qF2SxodXbBIY_DBuFWg,522
 truss/tests/util/test_env_vars.py,sha256=hthgB1mU0bJb1H4Jugc-0khArlLZ3x6tLE82cDaa-J0,390
 truss/tests/util/test_path.py,sha256=YfW3-IM_7iRsdR1Cb26KB1BkDsG_53_BUGBzoxY2Nog,7408
@@ -327,7 +330,6 @@ truss/truss_handle/patch/local_truss_patch_applier.py,sha256=fOHWKt3teYnbqeRsF63
 truss/truss_handle/patch/signature.py,sha256=8eas8gy6Japd1hrgdmtHmKTTxQmWsbmgKRQQGL2PVuA,858
 truss/truss_handle/patch/truss_dir_patch_applier.py,sha256=uhhHvKYHn_dpfz0xp4jwO9_qAej5sO3f8of_h-21PP4,3666
 truss/util/.truss_ignore,sha256=jpQA9ou-r_JEIcEHsUqGLHhir_m3d4IPGNyzKXtS-2g,3131
-truss/util/basetenpointer.py,sha256=PJ_meuTuXAopnWsHe1ZaH2RnltfmqdQ4QeXDQEPrblI,5596
 truss/util/docker.py,sha256=6PD7kMBBrOjsdvgkuSv7JMgZbe3NoJIeGasljMm2SwA,3934
 truss/util/download.py,sha256=1lfBwzyaNLEp7SAVrBd9BX5inZpkCVp8sBnS9RNoiJA,2521
 truss/util/env_vars.py,sha256=7Bv686eER71Barrs6fNamk_TrTJGmu9yV2TxaVmupn0,1232
@@ -357,12 +359,12 @@ truss_chains/remote_chainlet/model_skeleton.py,sha256=8ZReLOO2MLcdg7bNZ61C-6j-e6
 truss_chains/remote_chainlet/stub.py,sha256=Y2gDUzMY9WRaQNHIz-o4dfLUfFyYV9dUhIRQcfgrY8g,17209
 truss_chains/remote_chainlet/utils.py,sha256=O_5P-VAUvg0cegEW1uKCOf5EBwD8rEGYVoGMivOmc7k,22374
 truss_train/__init__.py,sha256=7hE6j6-u6UGzCGaNp3CsCN0kAVjBus1Ekups-Bk0fi4,837
-truss_train/definitions.py,sha256=yFQYJoxK2tDBeKFHR-IJz12jU1CtWRXN-ZERh9zjMHo,6622
-truss_train/deployment.py,sha256=zmeJ66kg1Wc7l7bwA_cXqv85uMF77hYl7NPHuhc1NPs,2493
-truss_train/loader.py,sha256=h-dDXao_J3cCXAxZvCqhAZHiYeQ5EhuA7NNE4fUSavk,2170
+truss_train/definitions.py,sha256=V985HhY4rdXL10DZxpFEpze9ScxzWErMht4WwaPknGU,6789
+truss_train/deployment.py,sha256=fDYRfzFRtVKMRVG0bKXYPmx6HXwLE0ukSQ0f81hG8kk,3020
+truss_train/loader.py,sha256=0o66EjBaHc2YY4syxxHVR4ordJWs13lNXnKjKq2wq0U,1630
 truss_train/public_api.py,sha256=9N_NstiUlmBuLUwH_fNG_1x7OhGCytZLNvqKXBlStrM,1220
-truss-0.10.9rc535.dist-info/METADATA,sha256=bSzdiExOCcAxu2hfSEmgkGFAt9fq4FOHoHsHl3H285Y,6674
-truss-0.10.9rc535.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-truss-0.10.9rc535.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
-truss-0.10.9rc535.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
-truss-0.10.9rc535.dist-info/RECORD,,
+truss-0.10.10rc0.dist-info/METADATA,sha256=WpmDKkfsnizWyPJCwbstGity9N5eI4KTyc1FOpEIIJk,6673
+truss-0.10.10rc0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+truss-0.10.10rc0.dist-info/entry_points.txt,sha256=-MwKfHHQHQ6j0HqIgvxrz3CehCmczDLTD-OsRHnjjuU,130
+truss-0.10.10rc0.dist-info/licenses/LICENSE,sha256=FTqGzu85i-uw1Gi8E_o0oD60bH9yQ_XIGtZbA1QUYiw,1064
+truss-0.10.10rc0.dist-info/RECORD,,

truss_train/definitions.py

@@ -18,6 +18,7 @@ class ModelWeightsFormat(str, enum.Enum):
 
     LORA = "lora"
     FULL = "full"
+    WHISPER = "whisper"
 
     def to_truss_config(self) -> "ModelWeightsFormat":
         return ModelWeightsFormat[self.name]
@@ -126,6 +127,7 @@ class TrainingJob(custom_types.SafeModelNoExtra):
     image: Image
     compute: Compute = Compute()
     runtime: Runtime = Runtime()
+    name: Optional[str] = None
 
     def model_dump(self, *args, **kwargs):
         data = super().model_dump(*args, **kwargs)
@@ -170,6 +172,10 @@ class FullCheckpoint(Checkpoint):
     model_weight_format: ModelWeightsFormat = ModelWeightsFormat.FULL
 
 
+class WhisperCheckpoint(Checkpoint):
+    model_weight_format: ModelWeightsFormat = ModelWeightsFormat.WHISPER
+
+
 class LoRACheckpoint(Checkpoint):
     lora_details: LoRADetails = LoRADetails()
     model_weight_format: ModelWeightsFormat = ModelWeightsFormat.LORA

truss_train/deployment.py

@@ -1,8 +1,9 @@
 import pathlib
 from pathlib import Path
-from typing import List
+from typing import List, Optional
 
 from truss.base.custom_types import SafeModel
+from truss.cli.utils.output import console
 from truss.remote.baseten import custom_types as b10_types
 from truss.remote.baseten.api import BasetenApi
 from truss.remote.baseten.core import archive_dir
@@ -44,6 +45,7 @@ def prepare_push(api: BasetenApi, config: pathlib.Path, training_job: TrainingJo
         image=training_job.image,
         runtime=training_job.runtime,
         compute=training_job.compute,
+        name=training_job.name,
         runtime_artifacts=[
             S3Artifact(s3_key=credentials["s3_key"], s3_bucket=credentials["s3_bucket"])
         ],
@@ -57,14 +59,25 @@ def create_training_job(
         training_project=training_project
     )
     prepared_job = prepare_push(remote_provider.api, config, training_project.job)
+
     job_resp = remote_provider.api.create_training_job(
         project_id=project_resp["id"], job=prepared_job
     )
     return job_resp
 
 
-def create_training_job_from_file(remote_provider: BasetenRemote, config: Path) -> dict:
+def create_training_job_from_file(
+    remote_provider: BasetenRemote,
+    config: Path,
+    job_name_from_cli: Optional[str] = None,
+) -> dict:
     with loader.import_training_project(config) as training_project:
+        if job_name_from_cli:
+            if training_project.job.name:
+                console.print(
+                    f"[bold yellow]⚠ Warning:[/bold yellow] name '{training_project.job.name}' provided in config file will be ignored. Using job name '{job_name_from_cli}' provided via --job-name flag."
+                )
+            training_project.job.name = job_name_from_cli
         job_resp = create_training_job(
             remote_provider=remote_provider,
             training_project=training_project,

truss_train/loader.py

@@ -5,7 +5,6 @@ import pathlib
 from typing import Iterator, Type, TypeVar
 
 from truss_train import definitions
-from truss.base.truss_spec import TrussSpec
 
 T = TypeVar("T")
 
@@ -35,28 +34,16 @@ def import_target(module_path: pathlib.Path, target_type: Type[T]) -> Iterator[T
             f"the training configuration is defined."
         )
 
-    if module_path.suffix == ".yaml":
-        import yaml
-        with open(module_path, 'r') as f:
-            yaml_content = yaml.safe_load(f)
-        try:
-            parsed_object = target_type(**yaml_content)
-            target = [parsed_object]
-        except Exception as e:
-            raise ValueError(f"Failed to parse YAML file {module_path} as {target_type.__name__}: {e}")
-            
-    elif module_path.suffix == ".py":
-        spec = importlib.util.spec_from_file_location(module_name, module_path)
-        if not spec or not spec.loader:
-            raise ImportError(f"Could not import `{module_path}`. Check path.")
+    spec = importlib.util.spec_from_file_location(module_name, module_path)
+    if not spec or not spec.loader:
+        raise ImportError(f"Could not import `{module_path}`. Check path.")
 
-        module = importlib.util.module_from_spec(spec)
-        spec.loader.exec_module(module)
+    module = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(module)
 
-        module_vars = (getattr(module, name) for name in dir(module))
-        target = [sym for sym in module_vars if isinstance(sym, target_type)]
+    module_vars = (getattr(module, name) for name in dir(module))
+    target = [sym for sym in module_vars if isinstance(sym, target_type)]
 
-    print(f"Target: {target}")
     if len(target) == 0:
         raise ValueError(f"No `{target_type}` was found.")
     elif len(target) > 1:

truss/tests/util/test_basetenpointer.py

@@ -1,227 +0,0 @@
-import time
-from pathlib import Path
-from tempfile import TemporaryDirectory
-
-import pytest
-import requests
-from huggingface_hub.errors import HfHubHTTPError
-
-from truss.base.truss_config import ModelCache, ModelRepo
-from truss.util.basetenpointer import model_cache_hf_to_b10ptr
-
-
-def test_dolly_12b():
-    ModelCached = ModelCache(
-        [
-            dict(
-                repo_id="databricks/dolly-v2-12b",
-                revision="19308160448536e378e3db21a73a751579ee7fdd",
-                use_volume=True,
-                volume_folder="databricks_dolly_v2_12b",
-                runtime_secret_name="hf_access_token",
-            )
-        ]
-    )
-    for _ in range(2):
-        try:
-            bptr = model_cache_hf_to_b10ptr(ModelCached)
-            continue
-        # timeout by huggingface hub timeout error
-        except requests.exceptions.ReadTimeout as e:
-            # this is expected to timeout when the request takes too long
-            # due to the large size of the model
-            print("ReadTimeout Error: ", e)
-            pytest.skip(
-                "Skipping test due to ReadTimeout error from Hugging Face API, "
-                "this can happen for large models like Dolly-12b"
-            )
-        except HfHubHTTPError as e:
-            if e.response.status_code == 429:
-                pytest.skip("Hugging Face API rate limit exceeded")
-            raise
-    bptr_list = bptr.pointers
-    expected = [
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/.gitattributes",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:.gitattributes",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/.gitattributes",
-            "hashtype": "etag",
-            "hash": "c7d9f3332a950355d5a77d85000f05e6f45435ea",
-            "size": 1477,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/README.md",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:README.md",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/README.md",
-            "hashtype": "etag",
-            "hash": "2912eb39545af0367335cff448d07214519c5eed",
-            "size": 10746,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/config.json",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:config.json",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/config.json",
-            "hashtype": "etag",
-            "hash": "888c677eda015e2375fad52d75062d14b30ebad9",
-            "size": 818,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/instruct_pipeline.py",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:instruct_pipeline.py",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/instruct_pipeline.py",
-            "hashtype": "etag",
-            "hash": "f8b291569e936cf104f44d003f95451bf5e1f965",
-            "size": 9159,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/pytorch_model.bin",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:pytorch_model.bin",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/pytorch_model.bin",
-            "hashtype": "etag",
-            "hash": "19e10711310992c310c3775964c7635f4b28dd86587403e718c6d6d524a406a5",
-            "size": 23834965761,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/special_tokens_map.json",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:special_tokens_map.json",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/special_tokens_map.json",
-            "hashtype": "etag",
-            "hash": "ecc1ee07dec13ee276fa9f1b29a1078da3280a4d",
-            "size": 228,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/tokenizer.json",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:tokenizer.json",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/tokenizer.json",
-            "hashtype": "etag",
-            "hash": "22868c8caf99a303c1a44bfea98f20f4254fc0e5",
-            "size": 2114274,
-        },
-        {
-            "resolution": {
-                "url": "https://huggingface.co/databricks/dolly-v2-12b/resolve/19308160448536e378e3db21a73a751579ee7fdd/tokenizer_config.json",
-                "expiration_timestamp": 2373918212,
-            },
-            "uid": "databricks/dolly-v2-12b:19308160448536e378e3db21a73a751579ee7fdd:tokenizer_config.json",
-            "file_name": "/app/model_cache/databricks_dolly_v2_12b/tokenizer_config.json",
-            "hashtype": "etag",
-            "hash": "51e564ead5d28eebc74b25d86f0a694b7c7cc618",
-            "size": 449,
-        },
-    ]
-    assert len(bptr_list) == len(expected), (
-        f"Expected {len(expected)} but got {len(bptr_list)}"
-    )
-    for expected, actual in zip(expected, bptr_list):
-        assert expected["uid"] == actual.uid, (
-            f"Expected uid {expected['uid']} but got {actual.uid}"
-        )
-        assert expected["file_name"] == actual.file_name, (
-            f"Expected file_name {expected['file_name']} but got {actual.file_name}"
-        )
-        assert expected["hash"] == actual.hash, (
-            f"Expected hash {expected['hash']} but got {actual.hash}"
-        )
-        assert expected["size"] == actual.size, (
-            f"Expected size {expected['size']} but got {actual.size}"
-        )
-        assert expected["resolution"]["url"] == actual.resolution.url, (
-            f"Expected resolution url {expected['resolution']['url']} but got {actual.resolution.url}"
-        )
-        # 100 years or more ahead
-        assert (
-            actual.resolution.expiration_timestamp
-            >= time.time() + 20 * 365 * 24 * 60 * 60
-        ), (
-            f"Expected unix expiration timestamp to be at least 20 years ahead, but got {actual.resolution.expiration_timestamp}. "
-        )
-
-    # download first file and verify size
-    with TemporaryDirectory() as tmp:
-        # Get the first pointer (.gitattributes)
-        first_pointer = bptr_list[0]
-        tmp_path = Path(tmp) / "downloaded_file"
-
-        # Download the file
-        response = requests.get(first_pointer.resolution.url)
-        response.raise_for_status()
-
-        # Save the file
-        tmp_path.write_bytes(response.content)
-
-        # Verify file size matches metadata
-        actual_size = tmp_path.stat().st_size
-        assert actual_size == first_pointer.size, (
-            f"Downloaded file size {actual_size} does not match expected size {first_pointer.size}"
-        )
-
-
-def test_with_main():
-    # main should be resolved to 41dec486b25746052d3335decc8f5961607418a0
-    cache = ModelCache(
-        [
-            ModelRepo(
-                repo_id="intfloat/llm-retriever-base",
-                revision="main",
-                ignore_patterns=["*.json", "*.txt", "*.md", "*.bin", "*.model"],
-                volume_folder="mistral_demo",
-                use_volume=True,
-            )
-        ]
-    )
-    b10ptr = model_cache_hf_to_b10ptr(cache)
-    expected = {
-        "pointers": [
-            {
-                "resolution": {
-                    "url": "https://huggingface.co/intfloat/llm-retriever-base/resolve/41dec486b25746052d3335decc8f5961607418a0/.gitattributes",
-                    "expiration_timestamp": 4044816725,
-                },
-                "uid": "intfloat/llm-retriever-base:main:.gitattributes",
-                "file_name": "/app/model_cache/mistral_demo/.gitattributes",
-                "hashtype": "etag",
-                "hash": "a6344aac8c09253b3b630fb776ae94478aa0275b",
-                "size": 1519,
-                "runtime_secret_name": "hf_access_token",
-            },
-            {
-                "resolution": {
-                    "url": "https://huggingface.co/intfloat/llm-retriever-base/resolve/41dec486b25746052d3335decc8f5961607418a0/model.safetensors",
-                    "expiration_timestamp": 4044816725,
-                },
-                "uid": "intfloat/llm-retriever-base:main:model.safetensors",
-                "file_name": "/app/model_cache/mistral_demo/model.safetensors",
-                "hashtype": "etag",
-                "hash": "565dd4f1cc6318ccf07af8680c27fd935b3b56ca2684d1af58abcd4e8bf6ecfa",
-                "size": 437955512,
-                "runtime_secret_name": "hf_access_token",
-            },
-        ]
-    }
-    assert b10ptr.model_dump() == expected
-
-
-if __name__ == "__main__":
-    test_dolly_12b()
-    test_with_main()

truss/util/basetenpointer.py

@@ -1,160 +0,0 @@
-"""This file contains the utils to create a basetenpointer from a huggingface repo, which can be resolved at runtime."""
-
-import time
-from pathlib import Path
-from typing import TYPE_CHECKING, Optional
-
-import requests
-from huggingface_hub import hf_api, hf_hub_url
-from huggingface_hub.utils import filter_repo_objects
-from pydantic import BaseModel
-
-if TYPE_CHECKING:
-    from truss.base.truss_config import ModelCache
-
-
-# copied from: https://github.com/basetenlabs/baseten/blob/caeba66cd544a5152bb6a018d6ac2871814f327b/baseten_shared/baseten_shared/lms/types.py#L13
-class Resolution(BaseModel):
-    url: str
-    expiration_timestamp: int
-
-
-class BasetenPointer(BaseModel):
-    resolution: Optional[Resolution] = None
-    uid: str
-    file_name: str
-    hashtype: str
-    hash: str
-    size: int
-    runtime_secret_name: str = "hf_access_token"  # TODO: remove the default
-
-
-class BasetenPointerList(BaseModel):
-    pointers: list[BasetenPointer]
-
-
-def get_hf_metadata(api: "hf_api.HfApi", repo: str, revision: str, file: str):
-    url = hf_hub_url(repo_id=repo, revision=revision, filename=file)
-    meta = api.get_hf_file_metadata(url=url)
-    return {"etag": meta.etag, "location": meta.location, "size": meta.size, "url": url}
-
-
-def filter_repo_files(
-    files: list[str],
-    allow_patterns: Optional[list[str]],
-    ignore_patterns: Optional[list[str]],
-) -> list[str]:
-    return list(
-        filter_repo_objects(
-            items=files, allow_patterns=allow_patterns, ignore_patterns=ignore_patterns
-        )
-    )
-
-
-def metadata_hf_repo(
-    repo: str,
-    revision: str,
-    allow_patterns: Optional[list[str]] = None,
-    ignore_patterns: Optional[list[str]] = None,
-) -> dict[str, dict]:
-    """Lists all files, gathers metadata without downloading, just using the Hugging Face API.
-    Example:
-    [{'.gitattributes': HfFileMetadata(
-    commit_hash='07163b72af1488142a360786df853f237b1a3ca1',
-    etag='a6344aac8c09253b3b630fb776ae94478aa0275b',
-    location='https://huggingface.co/intfloat/e5-mistral-7b-instruct/resolve/main/.gitattributes',
-    url='https://huggingface.co/intfloat/e5-mistral-7b-instruct/resolve/main/.gitattributes',
-    size=1519)]
-    """
-    api = hf_api.HfApi()
-    model_info = api.model_info(repo_id=repo, revision=revision)
-    real_revision = model_info.sha
-    real_revision = real_revision or revision
-    if revision != real_revision:
-        print(
-            f"Warning: revision {revision} is moving, using {real_revision} instead. "
-            f"Please update your code to use `revision={real_revision}` instead otherwise you will keep moving. "
-        )
-    files: list[str] = api.list_repo_files(repo_id=repo, revision=real_revision)
-    files = filter_repo_files(
-        files, allow_patterns=allow_patterns, ignore_patterns=ignore_patterns
-    )
-
-    hf_files_meta = {
-        file: get_hf_metadata(api, repo, real_revision, file) for file in files
-    }
-
-    return hf_files_meta
-
-
-def model_cache_hf_to_b10ptr(cache: "ModelCache") -> BasetenPointerList:
-    """
-    Convert a ModelCache object to a BasetenPointer object.
-    """
-    assert cache.is_v2, "ModelCache is not v2"
-
-    basetenpointers: list[BasetenPointer] = []
-
-    for model in cache.models:
-        assert model.revision is not None, "ModelCache is not v2, revision is None"
-        exception = None
-        for _ in range(3):
-            try:
-                metadata_hf_repo_list = metadata_hf_repo(
-                    repo=model.repo_id,
-                    revision=model.revision,
-                    allow_patterns=model.allow_patterns,
-                    ignore_patterns=model.ignore_patterns,
-                )
-                break
-            except requests.exceptions.ReadTimeout as e:
-                # this is expected, sometimes huggingface hub times out
-                print("ReadTimeout Error: ", e)
-                time.sleep(5)
-                exception = e
-            except Exception as e:
-                raise e
-        else:
-            # if we get here, we have exhausted the retries
-            assert exception is not None, "ReadTimeout Error: " + str(exception)
-            raise exception
-        # convert the metadata to b10 pointer format
-        b10_pointer_list = [
-            BasetenPointer(
-                uid=f"{model.repo_id}:{model.revision}:{filename}",
-                file_name=(Path(model.runtime_path) / filename).as_posix(),
-                hashtype="etag",
-                hash=content["etag"],
-                size=content["size"],
-                runtime_secret_name=model.runtime_secret_name,
-                resolution=Resolution(
-                    url=content["url"],
-                    expiration_timestamp=int(
-                        4044816725  # 90 years in the future, hf does not expire. needs to be static, to have cache hits.
-                    ),
-                ),
-            )
-            for filename, content in metadata_hf_repo_list.items()
-        ]
-        basetenpointers.extend(b10_pointer_list)
-
-    return BasetenPointerList(pointers=basetenpointers)
-
-
-if __name__ == "__main__":
-    # example usage
-    from truss.base.truss_config import ModelCache, ModelRepo
-
-    cache = ModelCache(
-        [
-            ModelRepo(
-                repo_id="intfloat/llm-retriever-base",
-                revision="main",
-                ignore_patterns=["*.json", "*.txt", "*.md", "*.bin", "*.model"],
-                volume_folder="mistral_demo",
-                use_volume=True,
-            )
-        ]
-    )
-    b10ptr = model_cache_hf_to_b10ptr(cache)
-    print(b10ptr.model_dump_json())