flpelerin commited on
Commit
602d956
·
verified ·
1 Parent(s): c9340cb

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +10 -1
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