锦此
|
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 |
|
Awni Hannun
|
2146bcd7ee
|
Quantize embedding / Update quantize API (#680)
* more async eval
* quantize embedding / update quantize api
* more updates for quantize
* update for quantize embeddings
* update sd quant API
* update sdxl quants
* error for datasets < batch_size
* async
* fix config loading
* fix quant
* fix tests
* fix req
* remove lm head if tie weights is true
* fix test
|
2024-04-18 18:16:10 -07:00 |
|
Angelos Katharopoulos
|
3f3741d229
|
Fix requirements and image2image strength/steps mismatch (#585)
|
2024-03-14 12:22:54 -07:00 |
|
zweifisch
|
d0fa6cfcae
|
feat: stable-diffusion t2i add --seed (#558)
|
2024-03-10 06:12:54 -07:00 |
|
Angelos Katharopoulos
|
3a9e6c3f70
|
Stable diffusion XL (#516)
|
2024-03-08 10:24:19 -08:00 |
|
Awni Hannun
|
5aa652d3c2
|
remove simplify (#379)
|
2024-01-26 13:54:49 -08:00 |
|
Yousif
|
7575125d5d
|
Added lora support for Phi-2 (#302)
* Added lora support for Phi-2
* Added Phi-2 support in fuse and convert
* format + readme
---------
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-01-12 13:45:30 -08:00 |
|
Rishi Narang
|
a2bc8426f2
|
Update txt2image.py (#285)
added np alias
|
2024-01-10 09:31:59 -08:00 |
|
Awni Hannun
|
6759dfddf1
|
Fix SD image conversion (#266)
|
2024-01-09 08:41:31 -08:00 |
|
Awni Hannun
|
27c0a8c002
|
Add llms subdir + update README (#145)
* add llms subdir + update README
* nits
* use same pre-commit as mlx
* update readmes a bit
* format
|
2023-12-20 10:22:25 -08:00 |
|
Awni Hannun
|
31bc57c4ff
|
add copyright in source
|
2023-11-30 11:08:53 -08:00 |
|
Angelos Katharopoulos
|
b364cc56cd
|
Add the Llama and Stable Diffusion examples
|
2023-11-29 10:38:20 -08:00 |
|