Hugging Face model: LeonardoBenitez/demo-vision-unlearning-munba
Author: LeonardoBenitez
Created: 2025-11-09 15:51:49
Last modified: 2025-11-11 20:36:52
Likes: 0 | Downloads: 0
Tags: tensorboard, model-index, region:us
Card metadata:
hyperparameters: lora_r: 4
lora_alpha: 4
is_lora_negated: yes
seed: 42
model_name_or_path: CompVis/stable-diffusion-v1-4
dataset_forget_name: assets/datasets/imagenette_splits/n03028079/train_forget
dataset_retain_name: assets/datasets/imagenette_splits/n03028079/train_retain
image_column: image
caption_column: text
validation_prompt: An image of a church
num_validation_images: 1
validation_epochs: 1
resolution: 512
center_crop: no
random_flip: yes
dataloader_num_workers: 2
per_device_train_batch_size: 4
gradient_accumulation_steps: 1
num_train_epochs: 2
learning_rate: 0.0002
lr_scheduler_type: constant
should_log: yes
local_rank: -1
device: cuda
n_gpu: 1
output_dir: assets/models/munba_church_002
hub_model_id: LeonardoBenitez/demo-vision-unlearning-munba
logging_dir: logs
logging_steps: 20
save_strategy: epoch
save_total_limit: 2
gradient_checkpointing: no
enable_xformers_memory_efficient_attention: no
mixed_precision: no
allow_tf32: no
use_8bit_adam: no
report_to: tensorboard
compute_gradient_conflict: no
compute_runtimes: yes
max_train_steps: 424
lr_warmup_steps: 0
adam_beta1: 0.9
adam_beta2: 0.999
adam_weight_decay: 0.01
adam_epsilon: 1e-08
max_grad_norm: 1
checkpointing_steps: 10000
noise_offset: 0