Nick088 commited on
Commit
9e30ec0
·
verified ·
1 Parent(s): 61c5d7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,9 @@ def generate(
24
 
25
  if seed == 0:
26
  seed = random.randint(1, 100000)
 
27
  else:
28
- seed = seed
29
 
30
  generate_kwargs = dict(
31
  temperature=temperature,
@@ -34,7 +35,6 @@ def generate(
34
  top_k=top_k,
35
  repetition_penalty=repetition_penalty,
36
  do_sample=True,
37
- seed=seed,
38
  )
39
 
40
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
 
24
 
25
  if seed == 0:
26
  seed = random.randint(1, 100000)
27
+ torch.manual_seed(seed)
28
  else:
29
+ torch.manual_seed(seed)
30
 
31
  generate_kwargs = dict(
32
  temperature=temperature,
 
35
  top_k=top_k,
36
  repetition_penalty=repetition_penalty,
37
  do_sample=True,
 
38
  )
39
 
40
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)