zhzluke96 commited on
Commit
0129fb6
1 Parent(s): 353b2e3
Files changed (2) hide show
  1. modules/utils/SeedContext.py +6 -1
  2. requirements.txt +2 -1
modules/utils/SeedContext.py CHANGED
@@ -56,7 +56,12 @@ class SeedContext:
56
  def __enter__(self):
57
  self.state = (torch.get_rng_state(), random.getstate(), np.random.get_state())
58
 
59
- deterministic(self.seed)
 
 
 
 
 
60
 
61
  def __exit__(self, exc_type, exc_value, traceback):
62
  torch.set_rng_state(self.state[0])
 
56
  def __enter__(self):
57
  self.state = (torch.get_rng_state(), random.getstate(), np.random.get_state())
58
 
59
+ try:
60
+ deterministic(self.seed)
61
+ except Exception as e:
62
+ raise ValueError(
63
+ f"Seed must be an integer, but: <{type(self.seed)}> {self.seed}"
64
+ )
65
 
66
  def __exit__(self, exc_type, exc_value, traceback):
67
  torch.set_rng_state(self.state[0])
requirements.txt CHANGED
@@ -22,4 +22,5 @@ gradio
22
  emojiswitch
23
  python-dotenv
24
  zhon
25
- mistune
 
 
22
  emojiswitch
23
  python-dotenv
24
  zhon
25
+ mistune==3.0.2
26
+ cn2an