Update lsb_release -a bitsandbytes device_map={'': 0} rich, check nvidia-smi with subproces.run

app.py

@@ -52,6 +52,21 @@ from transformers.generation import GenerationConfig
 
 from example_list import css, example_list
 
+if True:
+    ret = sp.run("lsb_release -a", capture_output=1, check=0, shell=1, encoding='utf8')
+    if ret.stdout:
+        rich.print(ret.stdout)
+    if ret.stderr:
+        rich.print("[red bold]" + ret.stdout)
+
+    ret = sp.run("nvidia-smi", capture_output=1, check=0, shell=1, encoding='utf8')
+    if ret.stdout:
+        rich.print(ret.stdout)
+    if ret.stderr:
+        rich.print("[red bold]" + ret.stdout)
+    raise SystemExit("Interrupt by intentioin")
+
+
 if not torch.cuda.is_available():
     raise gr.Error("torch.cuda.is_available() is False, cant continue...")