liwenyun
/
CLIP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix dropout_p type for torchscript conversion

This commit is contained in:
ofernandesumojo 2023-08-28 15:16:10 -05:00 committed by GitHub
parent a1d071733d
commit 3aff74026b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clip/model.py
View File

@ -81,7 +81,7 @@ class AttentionPool2d(nn.Module):
bias_k=None,
bias_v=None,
add_zero_attn=False,
dropout_p=0,
dropout_p=0.0,
out_proj_weight=self.c_proj.weight,
out_proj_bias=self.c_proj.bias,
use_separate_proj_weight=True,