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

Fix CI

Browse Source
This commit is contained in:
henryruhs
2025-02-14 16:40:39 +01:00
parent 88c4e53192
commit e1e0c11bb5
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/models/embedding_converter.py
+2 -4
View File
@@ -1,7 +1,5 @@
import torch
from torch import nn
from ..types import VisionTensor
from torch import Tensor, nn
class EmbeddingConverter(nn.Module):
@@ -20,7 +18,7 @@ class EmbeddingConverter(nn.Module):
nn.Linear(1024, 512)
])
def forward(self, input_tensor : VisionTensor) -> VisionTensor:
def forward(self, input_tensor : Tensor) -> Tensor:
output_tensor = input_tensor / torch.norm(input_tensor)
for layer in self.layers[:-1]: