AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

quality

Browse Source
This commit is contained in:
Nathan Lambert
2022-10-26 16:05:51 -07:00
parent 741122e722
commit b98c62eb61
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/diffusers/models/unet_1d_blocks.py
View File

@@ -211,7 +211,7 @@ class UNetMidBlock1D(nn.Module):
super().__init__()
if num_layers < 1:
raise ValueError(f"UNetMidBlock1D requires added num_layers >= 1")
raise ValueError("UNetMidBlock1D requires added num_layers >= 1")
out_channels = in_channels if out_channels is None else out_channels
@@ -251,7 +251,7 @@ class AttnDownBlock1D(nn.Module):
super().__init__()
if num_layers < 1:
raise ValueError(f"AttnDownBlock1D requires added num_layers >= 1")
raise ValueError("AttnDownBlock1D requires added num_layers >= 1")
mid_channels = out_channels if mid_channels is None else mid_channels
@@ -288,7 +288,7 @@ class DownBlock1D(nn.Module):
def __init__(self, out_channels: int, in_channels: int, mid_channels: int = None, num_layers: int = 2):
super().__init__()
if num_layers < 1:
raise ValueError(f"DownBlock1D requires added num_layers >= 1")
raise ValueError("DownBlock1D requires added num_layers >= 1")
mid_channels = out_channels if mid_channels is None else mid_channels
@@ -318,7 +318,7 @@ class DownBlock1DNoSkip(nn.Module):
def __init__(self, out_channels: int, in_channels: int, mid_channels: int = None, num_layers: int = 2):
super().__init__()
if num_layers < 1:
raise ValueError(f"DownBlock1DNoSkip requires added num_layers >= 1")
raise ValueError("DownBlock1DNoSkip requires added num_layers >= 1")
mid_channels = out_channels if mid_channels is None else mid_channels
@@ -346,7 +346,7 @@ class AttnUpBlock1D(nn.Module):
def __init__(self, in_channels: int, out_channels: int, mid_channels: int = None, num_layers: int = 2):
super().__init__()
if num_layers < 1:
raise ValueError(f"AttnUpBlock1D requires added num_layers >= 1")
raise ValueError("AttnUpBlock1D requires added num_layers >= 1")
mid_channels = out_channels if mid_channels is None else mid_channels
@@ -386,7 +386,7 @@ class UpBlock1D(nn.Module):
def __init__(self, in_channels: int, out_channels: int, mid_channels: int = None, num_layers: int = 2):
super().__init__()
if num_layers < 1:
raise ValueError(f"UpBlock1D requires added num_layers >= 1")
raise ValueError("UpBlock1D requires added num_layers >= 1")
mid_channels = in_channels if mid_channels is None else mid_channels
@@ -419,7 +419,7 @@ class UpBlock1DNoSkip(nn.Module):
def __init__(self, in_channels: int, out_channels: int, mid_channels: int = None, num_layers: int = 2):
super().__init__()
if num_layers < 1:
raise ValueError(f"UpBlock1D requires added num_layers >= 1")
raise ValueError("UpBlock1D requires added num_layers >= 1")
mid_channels = in_channels if mid_channels is None else mid_channels