Spaces:
Runtime error
Runtime error
Updated app.py to increase max len of tokens to 200
Browse files
app.py
CHANGED
@@ -83,7 +83,7 @@ class Phi2ProjModel(nn.Module):
|
|
83 |
self.phi2_model = phi2_model
|
84 |
|
85 |
def forward(self, input_embed):
|
86 |
-
max_len =
|
87 |
output = self.phi2_model.generate(inputs_embeds=input_embed,
|
88 |
max_new_tokens=max_len,
|
89 |
return_dict_in_generate = True,
|
|
|
83 |
self.phi2_model = phi2_model
|
84 |
|
85 |
def forward(self, input_embed):
|
86 |
+
max_len = 200
|
87 |
output = self.phi2_model.generate(inputs_embeds=input_embed,
|
88 |
max_new_tokens=max_len,
|
89 |
return_dict_in_generate = True,
|