howard-hou commited on
Commit
9cb62bd
1 Parent(s): f58f98c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -110,6 +110,7 @@ def generate(
110
 
111
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
112
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
 
113
  print(f'{timestamp} - vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')
114
  del out
115
  del state
 
110
 
111
  gpu_info = nvmlDeviceGetMemoryInfo(gpu_h)
112
  timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
113
+ print('sampled tokens:', all_tokens)
114
  print(f'{timestamp} - vram {gpu_info.total} used {gpu_info.used} free {gpu_info.free}')
115
  del out
116
  del state