llama-stack-mirror/llama_stack
Botao Chen f450a0fd32
Change post training run.yaml inference config (#710)
## Context
Colab notebook provides some limited free T4 GPU. 

Making post training template e2e works with colab notebook T4 is
critical for early adoption of the stack post training apis. However, we
found that the existing LlamaModelParallelGenerator
(https://github.com/meta-llama/llama-stack/blob/main/llama_stack/providers/inline/inference/meta_reference/inference.py#L82)
in meta-reference inference implementation isn't compatible with T4
machine.

In this PR, We change to disable create_distributed_process_group for
inference api in post training run.yaml config and setup up the
distributed env variables in notebook
<img width="493" alt="Screenshot 2025-01-02 at 3 48 08 PM"
src="https://github.com/user-attachments/assets/dd159f70-4cff-475c-b459-1fc6e2c720ba"
/>

to make meta reference inference compatible with the free T4 machine

 ## test
Test with the WIP post training showcase colab notebook
https://colab.research.google.com/drive/1K4Q2wZq232_Bpy2ud4zL9aRxvCWAwyQs?usp=sharing
2025-01-03 08:37:48 -08:00
..
apis Import from the right path (#708) 2025-01-02 13:15:31 -08:00
cli [remove import *] clean up import *'s (#689) 2024-12-27 15:45:44 -08:00
distribution Redact sensitive information from configs when printing, etc. 2025-01-02 13:54:02 -08:00
providers [#432] Add Groq Provider - chat completions (#609) 2025-01-03 08:27:49 -08:00
scripts Fix to conda env build script 2024-12-17 12:19:34 -08:00
templates Change post training run.yaml inference config (#710) 2025-01-03 08:37:48 -08:00
__init__.py export LibraryClient 2024-12-13 12:08:00 -08:00