Use HF names for registering fireworks and together models

llama_stack/templates/fireworks/doc_template.md

@@ -20,7 +20,7 @@ The following environment variables can be configured:
 The following models are available by default:
 
 {% for model in default_models %}
-- `{{ model.model_id }}`
+- `{{ model.model_id }} ({{ model.provider_model_id }})`
 {% endfor %}
 {% endif %}
 

llama_stack/templates/fireworks/fireworks.py

@@ -6,6 +6,8 @@
 
 from pathlib import Path
 
+from llama_models.sku_list import all_registered_models
+
 from llama_stack.distribution.datatypes import ModelInput, Provider, ShieldInput
 from llama_stack.providers.remote.inference.fireworks import FireworksImplConfig
 from llama_stack.providers.remote.inference.fireworks.fireworks import MODEL_ALIASES
@@ -28,7 +30,16 @@ def get_distribution_template() -> DistributionTemplate:
         config=FireworksImplConfig.sample_run_config(),
     )
 
-    default_models = [ModelInput(model_id=m.provider_model_id) for m in MODEL_ALIASES]
+    core_model_to_hf_repo = {
+        m.descriptor(): m.huggingface_repo for m in all_registered_models()
+    }
+    default_models = [
+        ModelInput(
+            model_id=core_model_to_hf_repo[m.llama_model],
+            provider_model_id=m.provider_model_id,
+        )
+        for m in MODEL_ALIASES
+    ]
 
     return DistributionTemplate(
         name="fireworks",

llama_stack/templates/fireworks/run.yaml

@@ -45,41 +45,41 @@ metadata_store:
   db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/fireworks}/registry.db
 models:
 - metadata: {}
-  model_id: fireworks/llama-v3p1-8b-instruct
+  model_id: meta-llama/Llama-3.1-8B-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p1-8b-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p1-70b-instruct
+  model_id: meta-llama/Llama-3.1-70B-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p1-70b-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p1-405b-instruct
+  model_id: meta-llama/Llama-3.1-405B-Instruct-FP8
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p1-405b-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p2-1b-instruct
+  model_id: meta-llama/Llama-3.2-3B-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p2-1b-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p2-3b-instruct
+  model_id: meta-llama/Llama-3.2-11B-Vision-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p2-3b-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p2-11b-vision-instruct
+  model_id: meta-llama/Llama-3.2-11B-Vision-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p2-11b-vision-instruct
 - metadata: {}
-  model_id: fireworks/llama-v3p2-90b-vision-instruct
+  model_id: meta-llama/Llama-3.2-90B-Vision-Instruct
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-v3p2-90b-vision-instruct
 - metadata: {}
-  model_id: fireworks/llama-guard-3-8b
+  model_id: meta-llama/Llama-Guard-3-8B
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-guard-3-8b
 - metadata: {}
-  model_id: fireworks/llama-guard-3-11b-vision
+  model_id: meta-llama/Llama-Guard-3-11B-Vision
   provider_id: null
-  provider_model_id: null
+  provider_model_id: fireworks/llama-guard-3-11b-vision
 shields:
 - params: null
   shield_id: meta-llama/Llama-Guard-3-8B