CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove VisionTensor from Discriminator

Browse Source
This commit is contained in:
henryruhs
2025-02-22 12:19:44 +01:00
parent a6e1405c70
commit f482d46798
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/models/discriminator.py
+2 -3
View File
@@ -1,10 +1,9 @@
import configparser
from typing import List
from torch import nn
from torch import Tensor, nn
from ..networks.nld import NLD
from ..types import VisionTensor
CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')
@@ -31,7 +30,7 @@ class Discriminator(nn.Module):
return discriminators
def forward(self, input_tensor : VisionTensor) -> List[List[VisionTensor]]:
def forward(self, input_tensor : Tensor) -> List[List[Tensor]]:
temp_tensor = input_tensor
output_tensors = []