mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-04-19 15:56:37 +02:00
Henry Ruhs
2e6394565a
* add sync batchnorm * replace random.choice with hash * fifty percent reduction * fix discriminator input * restore dataset.py * Remove duplicates * add discriminator_ratio to config * fix onnx export bug: replace round() with int() * Fix embedding naming * Introduce ModelWithConfigCheckpoint callback (#86) * Fix dist ini * Style: Refactor typing and improve code clarity in training.py (#88) * Add type casting for trainer params * Add type casting for trainer params * Add type casting for trainer params * Remove inplace activations for torch.compile compatibility (#89) * Fix README * improvise with norm layers & weighted average * add skip layer * use gelu instead of leaky_relu * cleanup * cleanup * Update dependencies * Different defaults and enable validation * Different defaults and enable validation * Revert to higher batch size * Just use copy over copy2 --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> Co-authored-by: NeuroDonu <112660822+NeuroDonu@users.noreply.github.com> Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
11 lines
222 B
Plaintext
11 lines
222 B
Plaintext
--extra-index-url https://download.pytorch.org/whl/cu128
|
|
albumentations==2.0.8
|
|
lightning==2.5.5
|
|
onnx==1.18.0
|
|
onnxruntime==1.22.0
|
|
pytorch-msssim==1.0.0
|
|
torch==2.8.0
|
|
torchdata==0.11.0
|
|
torchvision==0.23.0
|
|
tensorboard==2.20.0
|