mini-omni-2

Sleeping

gpt-omni commited on Sep 4

Commit

0acf43a

•

1 Parent(s): 0530d2a

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -381,7 +381,6 @@ class OmniInference:
             download_model(ckpt_dir)
         self.fabric, self.model, self.text_tokenizer, self.snacmodel, self.whispermodel = load_model(ckpt_dir, device)
-    @spaces.GPU
     def warm_up(self, sample='./data/samples/output1.wav'):
         for _ in self.run_AT_batch_stream(sample):
             pass

             download_model(ckpt_dir)
         self.fabric, self.model, self.text_tokenizer, self.snacmodel, self.whispermodel = load_model(ckpt_dir, device)
     def warm_up(self, sample='./data/samples/output1.wav'):
         for _ in self.run_AT_batch_stream(sample):
             pass