CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-07-28 16:08:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Improve lot of types, imports and names

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:09 +01:00
parent e33bc0d52a
commit b6b4f9f65b
10 changed files with 64 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/models/embedding_converter.py
+1 -1
View File
@@ -1,5 +1,5 @@
import torch
import torch.nn as nn
from torch import nn
from ..types import VisionTensor
embedding_converter/src/training.py
+2 -2
View File
@@ -7,7 +7,7 @@ import torch
from pytorch_lightning import Trainer
from pytorch_lightning.callbacks import ModelCheckpoint
from pytorch_lightning.tuner.tuning import Tuner
from torch import Tensor
from torch import Tensor, nn
from torch.utils.data import DataLoader, Dataset, TensorDataset, random_split
from .models.embedding_converter import EmbeddingConverter
@@ -21,7 +21,7 @@ class EmbeddingConverterTrainer(pytorch_lightning.LightningModule):
def __init__(self) -> None:
super(EmbeddingConverterTrainer, self).__init__()
self.embedding_converter = EmbeddingConverter()
self.mse_loss = torch.nn.MSELoss()
self.mse_loss = nn.MSELoss()
def forward(self, source_embedding : Embedding) -> Embedding:
return self.embedding_converter(source_embedding)