Follow up after recent PRs
This commit is contained in:
+6
-3
@@ -201,7 +201,7 @@ def run():
|
|||||||
p=2,
|
p=2,
|
||||||
dim=1,
|
dim=1,
|
||||||
)
|
)
|
||||||
# we don't need the residuals after computing refusal directions
|
# We don't need the residuals after computing refusal directions.
|
||||||
del good_residuals, bad_residuals
|
del good_residuals, bad_residuals
|
||||||
empty_cache()
|
empty_cache()
|
||||||
|
|
||||||
@@ -420,8 +420,11 @@ def run():
|
|||||||
continue
|
continue
|
||||||
|
|
||||||
user = huggingface_hub.whoami(token)
|
user = huggingface_hub.whoami(token)
|
||||||
fullname = user.get("fullname", user.get("name", "Unknown"))
|
fullname = user.get(
|
||||||
email = user.get("email", "Not provided")
|
"fullname",
|
||||||
|
user.get("name", "unknown user"),
|
||||||
|
)
|
||||||
|
email = user.get("email", "no email found")
|
||||||
print(f"Logged in as [bold]{fullname} ({email})[/]")
|
print(f"Logged in as [bold]{fullname} ({email})[/]")
|
||||||
|
|
||||||
repo_id = questionary.text(
|
repo_id = questionary.text(
|
||||||
|
|||||||
@@ -48,6 +48,9 @@ def batchify(items: list[T], batch_size: int) -> list[list[T]]:
|
|||||||
|
|
||||||
|
|
||||||
def empty_cache():
|
def empty_cache():
|
||||||
|
# Collecting garbage is not an idempotent operation, and to avoid OOM errors,
|
||||||
|
# gc.collect() has to be called both before and after emptying the backend cache.
|
||||||
|
# See https://github.com/p-e-w/heretic/pull/17 for details.
|
||||||
gc.collect()
|
gc.collect()
|
||||||
|
|
||||||
if torch.cuda.is_available():
|
if torch.cuda.is_available():
|
||||||
|
|||||||
Reference in New Issue
Block a user