Commit cfc5dea
committed
fix(train): fall back to SageMakerPublicHub when model not found in private hub
When _get_hub_content_metadata fails to find a model in the configured
private hub (e.g. sdktest), retry with SageMakerPublicHub. This handles
models like meta-textgeneration-llama-3-2-1b-instruct that only exist
in the public hub.1 parent 8e060f6 commit cfc5dea
1 file changed
Lines changed: 24 additions & 8 deletions
Lines changed: 24 additions & 8 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
63 | 63 | | |
64 | 64 | | |
65 | 65 | | |
66 | | - | |
67 | | - | |
68 | | - | |
69 | | - | |
70 | | - | |
71 | | - | |
72 | | - | |
73 | | - | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
74 | 90 | | |
75 | 91 | | |
76 | 92 | | |
| |||
0 commit comments