phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-12 05:54:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

address comments, configure within docker file

Browse source
This commit is contained in:
Xi Yan 2024-09-15 23:16:18 -07:00
parent fd67cfff39
commit e466ec389b
7 changed files with 53 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

21
llama_toolchain/configs/distributions/generate_distribution_specs.py
View file

@ -1,21 +0,0 @@
from llama_toolchain.core.distribution_registry import *
import json
import fire
import yaml
from llama_toolchain.common.serialize import EnumEncoder
def main():
for d in available_distribution_specs():
file_path = "./configs/distributions/distribution_registry/{}.yaml".format(
d.distribution_type
)
with open(file_path, "w") as f:
to_write = json.loads(json.dumps(d.dict(), cls=EnumEncoder))
f.write(yaml.dump(to_write, sort_keys=False))
if __name__ == "__main__":
fire.Fire(main)