CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-06-25 07:59:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Adjust like reviewed

Browse Source
This commit is contained in:
henryruhs
2025-03-29 09:56:23 +01:00
parent 4b851a173d
commit 7845dd8522
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/README.md
+1 -1
View File
@@ -112,7 +112,7 @@ target_path = .exports/face_swapper.onnx
target_size = 256
ir_version = 10
opset_version = 15
precision = fp16
precision = full
```
```
face_swapper/src/exporting.py
+3 -3
View File
@@ -12,7 +12,7 @@ CONFIG_PARSER = ConfigParser()
CONFIG_PARSER.read('config.ini')
class HalfPrecisionModel(nn.Module):
class HalfPrecision(nn.Module):
def __init__(self, model : Module) -> None:
super().__init__()
self.model = model.half()
@@ -38,8 +38,8 @@ def export() -> None:
os.makedirs(config_directory_path, exist_ok = True)
model = FaceSwapperTrainer.load_from_checkpoint(config_source_path, config_parser = CONFIG_PARSER, map_location = 'cpu').eval()
if config_precision == 'fp16':
model = HalfPrecisionModel(model).eval()
if config_precision == 'half':
model = HalfPrecision(model).eval()
model.ir_version = torch.tensor(config_ir_version)
source_tensor = torch.randn(1, 512)