锦此
|
f7bbe458ae
|
Add timeout to generate functions
Add timeout handling to various `generate` functions across multiple files.
* **cvae/main.py**
- Add `timeout` parameter to `generate` function.
- Implement timeout handling using `signal` module in `generate` function.
* **flux/dreambooth.py**
- Add `timeout` parameter to `generate_progress_images` function.
- Implement timeout handling using `signal` module in `generate_progress_images` function.
* **musicgen/generate.py**
- Add `timeout` parameter to `main` function.
- Implement timeout handling using `signal` module in `main` function.
* **stable_diffusion/txt2image.py**
- Add `timeout` parameter to `main` function.
- Implement timeout handling using `signal` module in `main` function.
* **llava/generate.py**
- Add `timeout` parameter to `main` function.
- Implement timeout handling using `signal` module in `main` function.
* **llms/gguf_llm/generate.py**
- Add `timeout` parameter to `generate` function.
- Implement timeout handling using `signal` module in `generate` function.
* **llms/mlx_lm/generate.py**
- Add `timeout` parameter to `main` function.
- Implement timeout handling using `signal` module in `main` function.
---
For more details, open the [Copilot Workspace session](https://copilot-workspace.githubnext.com/jincdream/mlx-examples?shareId=XXXX-XXXX-XXXX-XXXX).
|
2024-10-22 17:06:58 +08:00 |
|
Noah Kasmanoff
|
a429263905
|
LlaVA in MLX (#461)
* add: llava mlx first draft
* add: weights comparision
* add forward pass skeleton
* update: now imports weights correctly
* delete base
* latest
* adding config
* fix: use config
* add mlx config
* feat: add image processor for llava processor
* wip
* feat: llava working example
* chore: refactor generate script
* chore: clean up
* add: warning to user if no <image> token despite using one
* add: __call__ to LlavaModel
* add: call to LlavaModel
* update fp
* clean up var names
* update: native GeLU
* Cleanup
* update generate and readme
* remove todo comment
* rearrange tests
* fix example code
* nits in README
* update readme
* nit in readme
* nits in README
* chore(llava): refactor image embedding merging logic
* min mlx version
* nits in readmes
* fix cli prompt, some nits
* updates, slight simplify
---------
Co-authored-by: anchen <li.anchen.au@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-03-01 10:28:35 -08:00 |
|