Spaces:
Runtime error
Runtime error
TThis should work
Browse files
app.py
CHANGED
|
@@ -179,7 +179,7 @@ model.to(device)
|
|
| 179 |
context = torch.zeros((1, 1), dtype=torch.long, device=device)
|
| 180 |
|
| 181 |
def greet(start_character, number_of_tokens):
|
| 182 |
-
context[0][0] = encode(start_character)
|
| 183 |
max_new_tokens = number_of_tokens
|
| 184 |
return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
|
| 185 |
|
|
|
|
| 179 |
context = torch.zeros((1, 1), dtype=torch.long, device=device)
|
| 180 |
|
| 181 |
def greet(start_character, number_of_tokens):
|
| 182 |
+
context[0][0] = encode(start_character)[0]
|
| 183 |
max_new_tokens = number_of_tokens
|
| 184 |
return decode(model.generate(context, max_new_tokens=max_new_tokens)[0].tolist())
|
| 185 |
|