mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
Remove deprecated argument
This commit is contained in:
@@ -60,7 +60,7 @@ class UNet(nn.Module):
|
||||
class UNetPro(UNet):
|
||||
def __init__(self) -> None:
|
||||
super(UNet, self).__init__()
|
||||
self.resnet = models.resnet34(pretrained = True)
|
||||
self.resnet = models.resnet34()
|
||||
self.down_samples = self.create_down_samples(self)
|
||||
self.up_samples = self.create_up_samples()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user