Spaces:
Running
Running
debug
Browse files
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')
|