Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(),
|