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

Fix Generator

Browse Source
This commit is contained in:
henryruhs
2025-02-22 10:43:08 +01:00
parent dc0abff0ce
commit 206a1411d1
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/generator.py
+1 -1
View File
@@ -13,7 +13,7 @@ CONFIG.read('config.ini')
class Generator(nn.Module):
def __init__(self) -> None:
super(Generator, self).__init__()
encoder_type = CONFIG.getint('training.model.generator', 'encoder_type')
encoder_type = CONFIG.get('training.model.generator', 'encoder_type')
id_channels = CONFIG.getint('training.model.generator', 'id_channels')
num_blocks = CONFIG.getint('training.model.generator', 'num_blocks')
face_swapper/src/networks/unet.py
+1 -1
View File
@@ -59,7 +59,7 @@ class UNet(nn.Module):
class UNetPro(UNet):
def __init__(self) -> None:
super(UNetPro, self).__init__()
super(UNet, self).__init__()
self.resnet = models.resnet34(pretrained = True)
self.down_samples = self.create_down_samples(self)
self.up_samples = self.create_up_samples()