锦此
|
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 |
|