diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-13 07:14:24 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-13 07:14:24 +0200 |
| commit | a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e (patch) | |
| tree | 6a695b2b5a73cebc35ff9e581c70f1a0e75b62e8 /models/sparse.py | |
| parent | Experimental convnext discriminator support (diff) | |
| download | textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.tar.gz textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.tar.bz2 textual-inversion-diff-a0b63ee7f4a8c793c0d200c86ef07677aa4cbf2e.zip | |
Update
Diffstat (limited to 'models/sparse.py')
| -rw-r--r-- | models/sparse.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/sparse.py b/models/sparse.py index bcb2897..07b3413 100644 --- a/models/sparse.py +++ b/models/sparse.py | |||
| @@ -15,7 +15,7 @@ class PseudoSparseEmbedding(nn.Module): | |||
| 15 | if dropout_p > 0.0: | 15 | if dropout_p > 0.0: |
| 16 | self.dropout = nn.Dropout(p=dropout_p) | 16 | self.dropout = nn.Dropout(p=dropout_p) |
| 17 | else: | 17 | else: |
| 18 | self.dropout = lambda x: x | 18 | self.dropout = nn.Identity() |
| 19 | 19 | ||
| 20 | self.register_buffer('mapping', torch.zeros(0, device=device, dtype=torch.long)) | 20 | self.register_buffer('mapping', torch.zeros(0, device=device, dtype=torch.long)) |
| 21 | 21 | ||