finalf0 commited on
Commit
9564e60
1 Parent(s): 7544860
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -26,6 +26,10 @@ import modelscope_studio as mgr
26
  # For Mac with MPS (Apple silicon or AMD GPUs).
27
  # PYTORCH_ENABLE_MPS_FALLBACK=1 python web_demo_2.6.py --device mps
28
 
 
 
 
 
29
  # Argparser
30
  parser = argparse.ArgumentParser(description='demo')
31
  parser.add_argument('--device', type=str, default='cuda', help='cuda or mps')
 
26
  # For Mac with MPS (Apple silicon or AMD GPUs).
27
  # PYTORCH_ENABLE_MPS_FALLBACK=1 python web_demo_2.6.py --device mps
28
 
29
+ os.system("pip list|grep torch")
30
+ os.system("pip list|grep trans")
31
+ os.system("pip list|grep flash")
32
+
33
  # Argparser
34
  parser = argparse.ArgumentParser(description='demo')
35
  parser.add_argument('--device', type=str, default='cuda', help='cuda or mps')