dx2102 commited on
Commit
c6074a4
·
verified ·
1 Parent(s): d3bf393

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,6 +35,7 @@ cpu_pipe = transformers.pipeline(
35
  )
36
  # print devices
37
  print(f"{gpu_pipe.device = }, {cpu_pipe.device = }")
 
38
  print('Done')
39
 
40
  example_prefix = '''pitch duration wait velocity instrument
@@ -199,7 +200,7 @@ CPUs will be slower but there is no time limit.
199
  def end(self):
200
  queue.put(None)
201
  def background_fn():
202
- print(f"{pipe.device = })")
203
  result = pipe(
204
  prefix,
205
  streamer=MyStreamer(),
 
35
  )
36
  # print devices
37
  print(f"{gpu_pipe.device = }, {cpu_pipe.device = }")
38
+ print(f"{gpu_pipe.model.device = }, {cpu_pipe.model.device = }")
39
  print('Done')
40
 
41
  example_prefix = '''pitch duration wait velocity instrument
 
200
  def end(self):
201
  queue.put(None)
202
  def background_fn():
203
+ print(f"{pipe.device = }, {pipe.model.device = }")
204
  result = pipe(
205
  prefix,
206
  streamer=MyStreamer(),