oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-04-20 22:13:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

Improve SuperboogaV2 with Date/Time Embeddings, GPU Support, and Multiple File Formats (#6748)

Browse source
This commit is contained in:
Alireza Ghasemi 2025-02-18 02:38:15 +01:00 • committed by GitHub
parent 12f6f7ba9f
commit 01f20d2d9f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 227 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

6
extensions/superboogav2/optimize.py
View file

@ -39,11 +39,11 @@ def _markdown_hyperparams():
# Convert numpy types to python types.
def _convert_np_types(params):
for key in params:
if type(params[key]) == np.bool_:
if isinstance(params[key], np.bool_):
params[key] = bool(params[key])
elif type(params[key]) == np.int64:
elif isinstance(params[key], np.int64):
params[key] = int(params[key])
elif type(params[key]) == np.float64:
elif isinstance(params[key], np.float64):
params[key] = float(params[key])
return params