ffreemt commited on
Commit
f8bc02a
·
1 Parent(s): ca48eff

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

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -53,12 +53,14 @@ from transformers.generation import GenerationConfig
53
  from example_list import css, example_list
54
 
55
  if True:
 
56
  ret = sp.run("lsb_release -a", capture_output=1, check=0, shell=1, encoding='utf8')
57
  if ret.stdout:
58
  rich.print(ret.stdout)
59
  if ret.stderr:
60
  rich.print("[red bold]" + ret.stdout)
61
 
 
62
  ret = sp.run("nvidia-smi", capture_output=1, check=0, shell=1, encoding='utf8')
63
  if ret.stdout:
64
  rich.print(ret.stdout)
 
53
  from example_list import css, example_list
54
 
55
  if True:
56
+ logger.info("lsb_release -a")
57
  ret = sp.run("lsb_release -a", capture_output=1, check=0, shell=1, encoding='utf8')
58
  if ret.stdout:
59
  rich.print(ret.stdout)
60
  if ret.stderr:
61
  rich.print("[red bold]" + ret.stdout)
62
 
63
+ logger.info("nvidia-smi")
64
  ret = sp.run("nvidia-smi", capture_output=1, check=0, shell=1, encoding='utf8')
65
  if ret.stdout:
66
  rich.print(ret.stdout)