Fabrice-TIERCELIN commited on
Commit
9e15cd8
·
verified ·
1 Parent(s): 6f1239e

Readd initialization

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -94,13 +94,13 @@ class Tango:
94
  return outputs
95
  return list(self.chunks(outputs, samples))
96
 
97
- ## Initialize TANGO
98
- #
99
- #tango = Tango(device = "cpu")
100
- #tango.vae.to(device_type)
101
- #tango.stft.to(device_type)
102
- #tango.model.to(device_type)
103
- #
104
  #def update_seed(is_randomize_seed, seed):
105
  # if is_randomize_seed:
106
  # return random.randint(0, max_64_bit_int)
 
94
  return outputs
95
  return list(self.chunks(outputs, samples))
96
 
97
+ # Initialize TANGO
98
+
99
+ tango = Tango(device = "cpu")
100
+ tango.vae.to(device_type)
101
+ tango.stft.to(device_type)
102
+ tango.model.to(device_type)
103
+
104
  #def update_seed(is_randomize_seed, seed):
105
  # if is_randomize_seed:
106
  # return random.randint(0, max_64_bit_int)