From 3aff74026bd5353088e4b99ce886184815fbb223 Mon Sep 17 00:00:00 2001
From: ofernandesumojo <86738562+ofernandesumojo@users.noreply.github.com>
Date: Mon, 28 Aug 2023 15:16:10 -0500
Subject: [PATCH] Fix dropout_p type for torchscript conversion

---
 clip/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/clip/model.py b/clip/model.py
index 232b779..9bfd1f0 100644
--- a/clip/model.py
+++ b/clip/model.py
@@ -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,