CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

changes

Browse Source
This commit is contained in:
harisreedhar
2025-03-23 18:48:45 +05:30
parent 602e890af2
commit 99a8527e24
2 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/exporting.py
-1
View File
@@ -1,7 +1,6 @@
import os
from configparser import ConfigParser
import torch
from .training import FaceSwapperTrainer
face_swapper/src/inferencing.py
+2 -5
View File
@@ -4,7 +4,7 @@ import torch
from torchvision import io
from .helper import calc_embedding
from .models.generator import Generator
from .training import FaceSwapperTrainer
CONFIG_PARSER = configparser.ConfigParser()
CONFIG_PARSER.read('config.ini')
@@ -17,10 +17,7 @@ def infer() -> None:
config_target_path = CONFIG_PARSER.get('inferencing', 'target_path')
config_output_path = CONFIG_PARSER.get('inferencing', 'output_path')
state_dict = torch.load(config_generator_path).get('state_dict').get('generator')
generator = Generator(CONFIG_PARSER)
generator.load_state_dict(state_dict)
generator.eval()
generator = FaceSwapperTrainer.load_from_checkpoint(config_generator_path, config_parser = CONFIG_PARSER, map_location = 'cpu').eval()
embedder = torch.jit.load(config_embedder_path, map_location = 'cpu').eval()
source_tensor = io.read_image(config_source_path)