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 CI

Browse Source
This commit is contained in:
henryruhs
2025-03-06 09:28:19 +01:00
parent 8f1f63f2ef
commit d944d95bfd
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+2 -2
View File
@@ -3,7 +3,7 @@ import os
from typing import Tuple
import torch
from lightning import Trainer, LightningModule
from lightning import LightningModule, Trainer
from lightning.pytorch.callbacks import ModelCheckpoint
from lightning.pytorch.loggers import TensorBoardLogger
from torch import Tensor, nn
@@ -12,7 +12,7 @@ from torchdata.stateful_dataloader import StatefulDataLoader
from .dataset import StaticDataset
from .models.embedding_converter import EmbeddingConverter
from .types import Batch, Embedding, OptimizerSet, Config
from .types import Batch, Config, Embedding, OptimizerSet
CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')
face_swapper/src/training.py
+1 -1
View File
@@ -5,7 +5,7 @@ from typing import Tuple, cast
import torch
import torchvision
from lightning import Trainer, LightningModule
from lightning import LightningModule, Trainer
from lightning.pytorch.callbacks import ModelCheckpoint
from lightning.pytorch.loggers import TensorBoardLogger
from torch import Tensor, nn