Update util.py
Browse files
util.py
CHANGED
|
@@ -42,4 +42,13 @@ def generate_text(model, tokenizer, prompt: torch.Tensor, seq_len: int):
|
|
| 42 |
token = tokenizer.decode(token.item())
|
| 43 |
text += token
|
| 44 |
|
| 45 |
-
return text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
token = tokenizer.decode(token.item())
|
| 43 |
text += token
|
| 44 |
|
| 45 |
+
return text
|
| 46 |
+
|
| 47 |
+
|
| 48 |
+
|
| 49 |
+
def generate_name():
|
| 50 |
+
prefix = "mingru"
|
| 51 |
+
random_number = random.randint(0, 0xFFFF)
|
| 52 |
+
hex_code = f"{random_number:04x}"
|
| 53 |
+
unique_name = f"{prefix}-{hex_code}"
|
| 54 |
+
return unique_name
|