Commit Graph

92 Commits

Author SHA1 Message Date
filipstrand
1d8714d3b1 Add some example runs 2024-09-05 23:21:48 +02:00
filipstrand
a0ee9a7a8a Rename apply-lora arg 2024-09-05 21:50:31 +02:00
filipstrand
96848e320f Add support for multiple LoRAs again 2024-09-05 19:54:46 +02:00
filipstrand
ff9f998638 Small fixes 2024-09-04 23:16:02 +02:00
Filip Strand
d5f5f9607e
Fix type
Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-04 23:14:38 +02:00
filipstrand
f28fffe0cf Point out the .safetensors file again 2024-09-04 22:44:11 +02:00
filipstrand
a3567bddd7 Start supporting only single LoRA (and some other simplifications) 2024-09-04 22:15:03 +02:00
filipstrand
6d35db5e9a Use mx.load instead of safe_open 2024-09-04 21:05:19 +02:00
filipstrand
5ad5352481 Introduce weight util for shared methods 2024-09-04 20:20:27 +02:00
filipstrand
af41b08d90 Move LoRA related things to a LoraUtil (have to think about where to best place this later...) 2024-09-04 20:10:45 +02:00
filipstrand
8983070728 Code style improvements and some restructuring 2024-09-04 20:10:45 +02:00
Filip Strand
b4c27a6fab
raise error if layer cannot be loaded
Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-04 19:43:32 +02:00
elitexp
0fd03c7552
@Xuzzo - Weight Adjustment as Original Transformer
Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-04 21:52:51 +05:45
Sujip Maharjan
8d8812675e feat: --apply-lora a.safetensors b.safetensors c.safetensor --lora-scales 0.5 1.0 0.5 2024-09-03 19:36:15 +05:45
Sujip Maharjan
03b929e732 feat: Add --lora-scale parameter to adjust the scaling of LoRA weights 2024-09-03 18:39:01 +05:45
elitexp
4dfb3aa892
Clean up weight handler
@xuzzo Cleanup code

Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-03 18:19:21 +05:45
Sujip Maharjan
f82caa3298 Icrecream dependency removed, not required -- error 2024-09-03 13:16:35 +05:45
Sujip Maharjan
cd54ad137c Apply lora with --apply-lora <path-to-safetensors> 2024-09-03 07:38:04 +05:45
Filip Strand
94920896e8
Merge pull request #24 from filipstrand/basic-quantization-support
Basic quantization support
2024-08-30 22:24:05 +02:00
filipstrand
fdc4cb9892 Rewrite weight handling and add quantization support 2024-08-30 22:13:36 +02:00
filipstrand
478fc4f8bc Update readme with new TODO list 2024-08-23 11:54:21 +02:00
filipstrand
bf26270695 Clarify where models are downloaded 2024-08-23 10:27:45 +02:00
filipstrand
1254f3827c Update readme with new performance numbers 2024-08-21 21:22:16 +02:00
filipstrand
58700ebed3 Update readme with new performance numbers 2024-08-21 11:11:36 +02:00
filipstrand
580817afb6 Link to issue tracker for downloading the dev model 2024-08-21 09:56:11 +02:00
filipstrand
336d8a17da Add missing guidance flag in readme description 2024-08-21 09:10:23 +02:00
filipstrand
1f54b94175 Simplify: Remove latent creator class and inline logic 2024-08-21 00:48:08 +02:00
filipstrand
45104ebf48 Fix width property 2024-08-20 23:22:25 +02:00
filipstrand
c6f0a17d34 Once again ignore png and jpg 2024-08-20 22:58:10 +02:00
filipstrand
d0970162a5 Fix images in readme 2024-08-20 22:56:57 +02:00
filipstrand
04c2aaac26 Add helpful hints when configuring which model to use in example 2024-08-20 22:46:31 +02:00
filipstrand
7a277fc326 Rename flux_1_schnell folder to flux_1 2024-08-20 22:37:32 +02:00
Filip Strand
09cddd05e8
Merge pull request #10 from Xuzzo/feature/add_flux1dev
Include FLUX.1-Dev
2024-08-20 22:24:20 +02:00
filipstrand
ae99420f95 Merge remote-tracking branch 'refs/remotes/origin/main' into feature/add_flux1dev 2024-08-20 22:21:33 +02:00
filipstrand
01c8a372e6 Update readme with dev model description 2024-08-20 21:51:34 +02:00
filipstrand
43621decca Update readme with new performance numbers (and reformat the table) 2024-08-20 18:43:37 +02:00
filipstrand
f9f032a220 Update flag description 2024-08-20 18:17:33 +02:00
filipstrand
e3d2ee9092 Merge remote-tracking branch 'refs/remotes/origin/main' into feature/add_flux1dev 2024-08-20 18:15:10 +02:00
filipstrand
d94c9006a5 Fix: Actually use the --output flag 2024-08-20 18:10:00 +02:00
filipstrand
4d613c7390 Move num_train_steps to ModelConfig 2024-08-20 11:45:36 +02:00
filipstrand
aeb377d900 Fix: Use the correct seed variable 2024-08-20 11:36:01 +02:00
filipstrand
7795d47cd3 Add a separate runtime config 2024-08-20 10:59:28 +02:00
filipstrand
4b42e50257 Add separate model config and other small updates 2024-08-20 09:44:09 +02:00
Fabio
c0cc8059e9 fix cli 2024-08-19 21:41:19 +02:00
Fabio
d1b4557d2e small fix to main 2024-08-19 21:23:32 +02:00
Fabio
76f5f01d4c Merge branch 'main' into feature/add_flux1dev 2024-08-19 21:15:12 +02:00
Fabio
3710f101e5 small fix and update readme 2024-08-19 21:09:41 +02:00
Fabio
e60cd79f0c make config static and freeze after initialisation 2024-08-19 20:41:11 +02:00
filipstrand
801b132667 Update readme with latest performance numbers 2024-08-19 19:19:12 +02:00
filipstrand
f21a074929 Save image using util method and update README with command line argument descriptions 2024-08-19 18:27:23 +02:00