temp commit

This commit is contained in:
Botao Chen 2024-12-16 19:04:47 -08:00
parent b2dbb5e3fe
commit 30f6eb282f
7 changed files with 16 additions and 15 deletions

View file

@ -3,10 +3,17 @@ image_name: experimental-post-training
docker_image: null
conda_env: experimental-post-training
apis:
- inference
- telemetry
- datasetio
- post_training
providers:
inference:
- provider_id: meta-reference-inference
provider_type: inline::meta-reference
config:
max_seq_len: 4096
checkpoint_dir: null
datasetio:
- provider_id: huggingface-0
provider_type: remote::huggingface
@ -24,11 +31,11 @@ metadata_store:
namespace: null
type: sqlite
db_path: ${env.SQLITE_STORE_DIR:~/.llama/distributions/meta-reference-gpu}/registry.db
models:
- metadata: {}
model_id: ${env.POST_TRAINING_MODEL}
provider_id: meta-reference-inference
provider_model_id: null
models: []
# - metadata: {}
# model_id: ${env.POST_TRAINING_MODEL}
# provider_id: meta-reference-inference
# provider_model_id: null
shields: []
memory_banks: []
datasets:

View file

@ -16,7 +16,7 @@ providers:
- provider_id: meta-reference-inference
provider_type: inline::meta-reference-quantized
config:
model: ${env.INFERENCE_MODEL}
model: ${env.INFERENCE_MODEL} # please make sure your inference model here is added as resource
max_seq_len: 4096
checkpoint_dir: ${env.INFERENCE_CHECKPOINT_DIR:null}
quantization: