NagisaNao commited on
Commit
a1d0e90
·
verified ·
1 Parent(s): 5f5ae65

Upload 16 files

Browse files

fix display count result

files_cells/notebooks/en/auto-cleaner_en.ipynb CHANGED
@@ -257,7 +257,7 @@
257
  " if file.endswith(\".txt\"):\n",
258
  " continue\n",
259
  " os.remove(os.path.join(root, file))\n",
260
- " if not file.endswith(\".yaml\"):\n",
261
  " deleted_files += 1\n",
262
  " return deleted_files\n",
263
  "\n",
 
257
  " if file.endswith(\".txt\"):\n",
258
  " continue\n",
259
  " os.remove(os.path.join(root, file))\n",
260
+ " if file.endswith((\".safetensors\", \".pt\")):\n",
261
  " deleted_files += 1\n",
262
  " return deleted_files\n",
263
  "\n",
files_cells/notebooks/ru/auto-cleaner_ru.ipynb CHANGED
@@ -257,7 +257,7 @@
257
  " if file.endswith(\".txt\"):\n",
258
  " continue\n",
259
  " os.remove(os.path.join(root, file))\n",
260
- " if not file.endswith(\".yaml\"):\n",
261
  " deleted_files += 1\n",
262
  " return deleted_files\n",
263
  "\n",
 
257
  " if file.endswith(\".txt\"):\n",
258
  " continue\n",
259
  " os.remove(os.path.join(root, file))\n",
260
+ " if file.endswith((\".safetensors\", \".pt\")):\n",
261
  " deleted_files += 1\n",
262
  " return deleted_files\n",
263
  "\n",
files_cells/python/en/auto-cleaner_en.py CHANGED
@@ -238,7 +238,7 @@ def clean_directory(directory):
238
  if file.endswith(".txt"):
239
  continue
240
  os.remove(os.path.join(root, file))
241
- if not file.endswith(".yaml"):
242
  deleted_files += 1
243
  return deleted_files
244
 
 
238
  if file.endswith(".txt"):
239
  continue
240
  os.remove(os.path.join(root, file))
241
+ if file.endswith((".safetensors", ".pt")):
242
  deleted_files += 1
243
  return deleted_files
244
 
files_cells/python/ru/auto-cleaner_ru.py CHANGED
@@ -238,7 +238,7 @@ def clean_directory(directory):
238
  if file.endswith(".txt"):
239
  continue
240
  os.remove(os.path.join(root, file))
241
- if not file.endswith(".yaml"):
242
  deleted_files += 1
243
  return deleted_files
244
 
 
238
  if file.endswith(".txt"):
239
  continue
240
  os.remove(os.path.join(root, file))
241
+ if file.endswith((".safetensors", ".pt")):
242
  deleted_files += 1
243
  return deleted_files
244