├── .github └── workflows │ ├── publish.yml │ └── smoketest.yml ├── .gitignore ├── LICENSE ├── NOTICES.md ├── README.md ├── docs └── static │ ├── architecture.png │ ├── image-1.png │ ├── image-10.png │ ├── image-11.png │ ├── image-2.png │ ├── image-3.png │ ├── image-4.png │ ├── image-5.png │ ├── image-6.png │ ├── image-7.png │ ├── image-8.png │ ├── image-9.png │ └── image.png ├── higgsfield ├── __init__.py ├── checkpoint │ ├── __init__.py │ ├── fsdp_checkpoint.py │ └── fsdp_utils.py ├── dataset │ ├── __init__.py │ ├── dataset.py │ └── openai.py ├── experiment.py ├── internal │ ├── __init__.py │ ├── cfg.py │ ├── ci │ │ ├── cli.py │ │ └── setup.py │ ├── cli.py │ ├── experiment │ │ ├── ast_parser.py │ │ ├── builder.py │ │ ├── decorator.py │ │ └── params.py │ ├── init.py │ ├── launch.py │ ├── main.py │ └── util.py ├── llama │ ├── __init__.py │ ├── llama.py │ └── llama_utils.py ├── loaders │ ├── __init__.py │ └── llama_loader.py ├── mistral │ ├── __init__.py │ ├── mistral.py │ ├── mistral_loader.py │ └── mistral_utils.py ├── path.py ├── rl │ ├── README.md │ └── rl_adventure_2 │ │ ├── 1.actor-critic.ipynb │ │ ├── 2.gae.ipynb │ │ ├── 3.ppo.ipynb │ │ ├── 4.acer.ipynb │ │ ├── 5.ddpg.ipynb │ │ ├── 6.td3.ipynb │ │ ├── 7.soft actor-critic.ipynb │ │ ├── 8.gail.ipynb │ │ ├── 9.her.ipynb │ │ ├── README.md │ │ └── common │ │ ├── __init__.py │ │ └── multiprocessing_env.py ├── static │ ├── project │ │ ├── .gitignore │ │ ├── Dockerfile │ │ ├── env │ │ ├── requirements.txt │ │ └── src │ │ │ ├── alpaca_bf16.py │ │ │ ├── alpaca_fp16.py │ │ │ └── dataset.py │ └── templates │ │ ├── README_md.j2 │ │ ├── config_py.j2 │ │ ├── deploy_action.j2 │ │ ├── experiment_action.j2 │ │ └── kill_action.j2 ├── training │ ├── __init__.py │ ├── grads.py │ └── scaler.py └── utils │ ├── __init__.py │ └── flush.py ├── poetry.lock ├── pyproject.toml ├── setup.md ├── tutorial.md └── tutorials ├── README.md ├── chatgpt.ipynb ├── prompt_completion.ipynb └── text_format.ipynb /.github/workflows/publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/.github/workflows/publish.yml -------------------------------------------------------------------------------- /.github/workflows/smoketest.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/.github/workflows/smoketest.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICES.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/NOTICES.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/README.md -------------------------------------------------------------------------------- /docs/static/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/architecture.png -------------------------------------------------------------------------------- /docs/static/image-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-1.png -------------------------------------------------------------------------------- /docs/static/image-10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-10.png -------------------------------------------------------------------------------- /docs/static/image-11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-11.png -------------------------------------------------------------------------------- /docs/static/image-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-2.png -------------------------------------------------------------------------------- /docs/static/image-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-3.png -------------------------------------------------------------------------------- /docs/static/image-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-4.png -------------------------------------------------------------------------------- /docs/static/image-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-5.png -------------------------------------------------------------------------------- /docs/static/image-6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-6.png -------------------------------------------------------------------------------- /docs/static/image-7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-7.png -------------------------------------------------------------------------------- /docs/static/image-8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-8.png -------------------------------------------------------------------------------- /docs/static/image-9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image-9.png -------------------------------------------------------------------------------- /docs/static/image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/docs/static/image.png -------------------------------------------------------------------------------- /higgsfield/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /higgsfield/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/checkpoint/__init__.py -------------------------------------------------------------------------------- /higgsfield/checkpoint/fsdp_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/checkpoint/fsdp_checkpoint.py -------------------------------------------------------------------------------- /higgsfield/checkpoint/fsdp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/checkpoint/fsdp_utils.py -------------------------------------------------------------------------------- /higgsfield/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/dataset/__init__.py -------------------------------------------------------------------------------- /higgsfield/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/dataset/dataset.py -------------------------------------------------------------------------------- /higgsfield/dataset/openai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/dataset/openai.py -------------------------------------------------------------------------------- /higgsfield/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/experiment.py -------------------------------------------------------------------------------- /higgsfield/internal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /higgsfield/internal/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/cfg.py -------------------------------------------------------------------------------- /higgsfield/internal/ci/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/ci/cli.py -------------------------------------------------------------------------------- /higgsfield/internal/ci/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/ci/setup.py -------------------------------------------------------------------------------- /higgsfield/internal/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/cli.py -------------------------------------------------------------------------------- /higgsfield/internal/experiment/ast_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/experiment/ast_parser.py -------------------------------------------------------------------------------- /higgsfield/internal/experiment/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/experiment/builder.py -------------------------------------------------------------------------------- /higgsfield/internal/experiment/decorator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/experiment/decorator.py -------------------------------------------------------------------------------- /higgsfield/internal/experiment/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/experiment/params.py -------------------------------------------------------------------------------- /higgsfield/internal/init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/init.py -------------------------------------------------------------------------------- /higgsfield/internal/launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/launch.py -------------------------------------------------------------------------------- /higgsfield/internal/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/main.py -------------------------------------------------------------------------------- /higgsfield/internal/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/internal/util.py -------------------------------------------------------------------------------- /higgsfield/llama/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/llama/__init__.py -------------------------------------------------------------------------------- /higgsfield/llama/llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/llama/llama.py -------------------------------------------------------------------------------- /higgsfield/llama/llama_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/llama/llama_utils.py -------------------------------------------------------------------------------- /higgsfield/loaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/loaders/__init__.py -------------------------------------------------------------------------------- /higgsfield/loaders/llama_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/loaders/llama_loader.py -------------------------------------------------------------------------------- /higgsfield/mistral/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /higgsfield/mistral/mistral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/mistral/mistral.py -------------------------------------------------------------------------------- /higgsfield/mistral/mistral_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/mistral/mistral_loader.py -------------------------------------------------------------------------------- /higgsfield/mistral/mistral_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/mistral/mistral_utils.py -------------------------------------------------------------------------------- /higgsfield/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/path.py -------------------------------------------------------------------------------- /higgsfield/rl/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/README.md -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/1.actor-critic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/1.actor-critic.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/2.gae.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/2.gae.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/3.ppo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/3.ppo.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/4.acer.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/4.acer.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/5.ddpg.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/5.ddpg.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/6.td3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/6.td3.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/7.soft actor-critic.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/7.soft actor-critic.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/8.gail.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/8.gail.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/9.her.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/9.her.ipynb -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/README.md -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/common/__init__.py: -------------------------------------------------------------------------------- 1 | import multiprocessing_env -------------------------------------------------------------------------------- /higgsfield/rl/rl_adventure_2/common/multiprocessing_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/rl/rl_adventure_2/common/multiprocessing_env.py -------------------------------------------------------------------------------- /higgsfield/static/project/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/.gitignore -------------------------------------------------------------------------------- /higgsfield/static/project/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/Dockerfile -------------------------------------------------------------------------------- /higgsfield/static/project/env: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/env -------------------------------------------------------------------------------- /higgsfield/static/project/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/requirements.txt -------------------------------------------------------------------------------- /higgsfield/static/project/src/alpaca_bf16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/src/alpaca_bf16.py -------------------------------------------------------------------------------- /higgsfield/static/project/src/alpaca_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/src/alpaca_fp16.py -------------------------------------------------------------------------------- /higgsfield/static/project/src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/project/src/dataset.py -------------------------------------------------------------------------------- /higgsfield/static/templates/README_md.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/templates/README_md.j2 -------------------------------------------------------------------------------- /higgsfield/static/templates/config_py.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/templates/config_py.j2 -------------------------------------------------------------------------------- /higgsfield/static/templates/deploy_action.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/templates/deploy_action.j2 -------------------------------------------------------------------------------- /higgsfield/static/templates/experiment_action.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/templates/experiment_action.j2 -------------------------------------------------------------------------------- /higgsfield/static/templates/kill_action.j2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/static/templates/kill_action.j2 -------------------------------------------------------------------------------- /higgsfield/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/training/__init__.py -------------------------------------------------------------------------------- /higgsfield/training/grads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/training/grads.py -------------------------------------------------------------------------------- /higgsfield/training/scaler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/training/scaler.py -------------------------------------------------------------------------------- /higgsfield/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import empty_cache -------------------------------------------------------------------------------- /higgsfield/utils/flush.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/higgsfield/utils/flush.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/setup.md -------------------------------------------------------------------------------- /tutorial.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/tutorial.md -------------------------------------------------------------------------------- /tutorials/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/tutorials/README.md -------------------------------------------------------------------------------- /tutorials/chatgpt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/tutorials/chatgpt.ipynb -------------------------------------------------------------------------------- /tutorials/prompt_completion.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/tutorials/prompt_completion.ipynb -------------------------------------------------------------------------------- /tutorials/text_format.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/higgsfield-ai/higgsfield/HEAD/tutorials/text_format.ipynb --------------------------------------------------------------------------------