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:25:38 +01:00
parent b59e172fa3
commit 8f1f63f2ef
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/training.py
+2 -3
View File
@@ -3,10 +3,9 @@ import os
import warnings
from typing import Tuple, cast
import lightning
import torch
import torchvision
from lightning import Trainer
from lightning import Trainer, LightningModule
from lightning.pytorch.callbacks import ModelCheckpoint
from lightning.pytorch.loggers import TensorBoardLogger
from torch import Tensor, nn
@@ -26,7 +25,7 @@ CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')
class FaceSwapperTrainer(lightning.LightningModule):
class FaceSwapperTrainer(LightningModule):
def __init__(self) -> None:
super().__init__()
embedder_path = CONFIG.get('training.model', 'embedder_path')