Commit
•
fb36b5a
1
Parent(s):
b84c226
Update clipGPT.py
Browse files- clipGPT.py +2 -2
clipGPT.py
CHANGED
|
@@ -70,11 +70,11 @@ 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=50,
|
| 77 |
-
temperature=0.
|
| 78 |
stop_token: str = ".",
|
| 79 |
):
|
| 80 |
|
|
|
|
| 70 |
def generate_beam(
|
| 71 |
model,
|
| 72 |
tokenizer,
|
| 73 |
+
beam_size: int = 13,
|
| 74 |
prompt=None,
|
| 75 |
embed=None,
|
| 76 |
entry_length=50,
|
| 77 |
+
temperature=0.9,
|
| 78 |
stop_token: str = ".",
|
| 79 |
):
|
| 80 |
|