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