Commit Graph

75 Commits

Author SHA1 Message Date
henryruhs 650551c19b Simplify Batch type 2025-03-11 14:43:09 +01:00
henryruhs a971506271 Revert changes in generator 2025-03-11 14:43:09 +01:00
henryruhs b69f69d015 Fix UNet 2025-03-11 14:43:09 +01:00
henryruhs 953525e6b0 Fix typo 2025-03-11 14:43:09 +01:00
henryruhs 58ad6af619 Add Sample suffix again 2025-03-11 14:43:09 +01:00
henryruhs 7264884ff9 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 1872f99584 Refacto UNet 2025-03-11 14:43:09 +01:00
henryruhs 29e82f909a Move nld to networks 2025-03-11 14:43:09 +01:00
henryruhs 257ab668ee Name what it is 2025-03-11 14:43:09 +01:00
henryruhs 62d897f9d8 Direct return layers 2025-03-11 14:43:09 +01:00
henryruhs f05ff6cdb1 Enforce IR version for older onnxruntime 2025-03-11 14:43:09 +01:00
henryruhs 3fe32d7832 Fix example config 2025-03-11 14:43:09 +01:00
henryruhs 3be8368eaa Improve lot of types, imports and names 2025-03-11 14:43:09 +01:00
henryruhs b6b4f9f65b Improve lot of types, imports and names 2025-03-11 14:43:09 +01:00
henryruhs e33bc0d52a Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs f3409c5ade Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs c6d16c0cf6 Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs dfe7ab3b6f Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs 3c6dfa4efe Modernize to use ModuleList, Fix some types 2025-03-11 14:43:09 +01:00
henryruhs 34d0bc10ed Fix append 2025-03-11 14:43:09 +01:00
henryruhs b785525f3b Fix append 2025-03-11 14:43:09 +01:00
henryruhs dd320ea5be Fix self 2025-03-11 14:43:09 +01:00
henryruhs 71e0ae34c0 Fix self 2025-03-11 14:43:09 +01:00
henryruhs 32dfdcf1b3 Remove unsued 2025-03-11 14:43:09 +01:00
henryruhs 494b84aecb Refactor discriminator to use ModuleList, Reduce complexity of layer creation 2025-03-11 14:43:09 +01:00
henryruhs 860771e482 Fix weight identity 2025-03-11 14:43:09 +01:00
henryruhs b42d2b06e7 Improve generate_preview 2025-03-11 14:43:09 +01:00
henryruhs 11c038cb81 Fix monitor 2025-03-11 14:43:09 +01:00
henryruhs 746cc86d52 Fix monitor 2025-03-11 14:43:09 +01:00
henryruhs 58c81cd646 Move magic methods up 2025-03-11 14:43:09 +01:00
henryruhs 4d2038d4ce Rename loss_id to loss_identity 2025-03-11 14:43:09 +01:00
henryruhs 67ad9badac Small typo 2025-03-11 14:43:09 +01:00
henryruhs 1f4405be44 Fix CI 2025-03-11 14:43:09 +01:00
henryruhs 1b6e7a6ca5 Rename ArcFace Converter to Embedding Converter, Add EmbeddingDataset, Add learning rate to config 2025-03-11 14:43:06 +01:00
henryruhs 62a69cddd2 Fix import order 2025-03-11 14:43:03 +01:00
henryruhs 611618e413 Make Embedding great again 2025-03-11 14:43:03 +01:00
harisreedhar 6381e755d7 list join 2025-03-11 14:43:03 +01:00
harisreedhar 2ed558a873 cleanup 2025-03-11 14:43:03 +01:00
henryruhs b7e2d3ccd7 It should be models and networks 2025-03-11 14:43:03 +01:00
harisreedhar 66af7d7957 add face swapper preview 2025-03-11 14:43:03 +01:00
harisreedhar dc0ef53668 add face swapper preview 2025-03-11 14:43:03 +01:00
henryruhs a17d050648 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 989a81c751 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs a1cd025f81 Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 9c15f584aa Mixed bag of cleanups 2025-03-11 14:43:03 +01:00
henryruhs 5892460c3d Add the final license 2025-03-11 14:43:03 +01:00
harisreedhar bb0e3b4a8a remove liveportrait and update requirements 2025-03-11 14:43:03 +01:00
harisreedhar fcb3390796 again again cleaning 2025-03-11 14:43:03 +01:00
harisreedhar 3b8b6442fc add infer and some cleaning 2025-03-11 14:43:03 +01:00
harisreedhar cfcf0ee2bd add glob image pattern to config 2025-03-11 14:43:03 +01:00