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

Modernize to use ModuleList, Fix some types

Browse Source
This commit is contained in:
henryruhs
2025-02-12 17:08:12 +01:00
parent c6d16c0cf6
commit f3409c5ade
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/models/embedding_converter.py
+2 -2
View File
@@ -1,7 +1,7 @@
import torch
import torch.nn as nn
from embedding_converter.src.types import VisionTensor
from ..types import VisionTensor
class EmbeddingConverter(nn.Module):
@@ -21,7 +21,7 @@ class EmbeddingConverter(nn.Module):
])
return layers
def forward(self, input_tensor: VisionTensor) -> VisionTensor:
def forward(self, input_tensor : VisionTensor) -> VisionTensor:
output_tensor = input_tensor / torch.norm(input_tensor)
for layer in self.layers[:-1]: