Update clipGPT.py
Browse files- clipGPT.py +2 -2
clipGPT.py
CHANGED
@@ -70,10 +70,10 @@ class ClipGPT2Model(nn.Module):
|
|
70 |
def generate_beam(
|
71 |
model,
|
72 |
tokenizer,
|
73 |
-
beam_size: int =
|
74 |
prompt=None,
|
75 |
embed=None,
|
76 |
-
entry_length=
|
77 |
temperature=0.9,
|
78 |
stop_token: str = ".",
|
79 |
):
|
|
|
70 |
def generate_beam(
|
71 |
model,
|
72 |
tokenizer,
|
73 |
+
beam_size: int = 10,
|
74 |
prompt=None,
|
75 |
embed=None,
|
76 |
+
entry_length=76,
|
77 |
temperature=0.9,
|
78 |
stop_token: str = ".",
|
79 |
):
|