Spaces:
Sleeping
Sleeping
howard-hou
commited on
Commit
•
9cb62bd
1
Parent(s):
f58f98c
Update app.py
Browse files
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
|