Commit
•
cd96f35
1
Parent(s):
fb36b5a
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 |
):
|