Update app.py

app.py

@@ -19,20 +19,18 @@ os.makedirs('./temp', exist_ok=True)
 
 print('\n\n\n')
 print('Loading model...')
-gpu_pipe = transformers.pipeline(
-    'text-generation', 
-    model='dx2102/llama-midi',
-    # revision='c303c108399aba837146e893375849b918f413b3',
-    torch_dtype='bfloat16',
-    device='cuda',
-)
 cpu_pipe = transformers.pipeline(
     'text-generation', 
     model='dx2102/llama-midi',
-    # revision='c303c108399aba837146e893375849b918f413b3',
     torch_dtype='float32',
     device='cpu',
 )
+gpu_pipe = transformers.pipeline(
+    'text-generation', 
+    model='dx2102/llama-midi',
+    torch_dtype='bfloat16',
+    device='cuda',
+)
 # print devices
 print(f"{gpu_pipe.device = }, {cpu_pipe.device = }")
 print(f"{gpu_pipe.model.device = }, {cpu_pipe.model.device = }")