mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
Fix CI
This commit is contained in:
@@ -11,7 +11,7 @@ class UNet(nn.Module):
|
||||
self.up = self.create_up()
|
||||
|
||||
@staticmethod
|
||||
def create_down():
|
||||
def create_down() -> nn.ModuleList:
|
||||
return nn.ModuleList(
|
||||
[
|
||||
Down(3, 32),
|
||||
@@ -24,7 +24,7 @@ class UNet(nn.Module):
|
||||
])
|
||||
|
||||
@staticmethod
|
||||
def create_up():
|
||||
def create_up() -> nn.ModuleList:
|
||||
return nn.ModuleList(
|
||||
[
|
||||
Up(1024, 1024),
|
||||
|
||||
Reference in New Issue
Block a user