YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
image_size = 128 # the generated image resolution train_batch_size = 16 eval_batch_size = 16 # how many images to sample during evaluation num_epochs = 15000 gradient_accumulation_steps = 1 learning_rate = 7e-5 lr_warmup_steps = 500 save_image_epochs = 50 save_model_epochs = 50
preprocess = transforms.Compose( [ transforms.Resize((config.image_size, config.image_size)), transforms.RandomRotation(5), # transforms.RandomHorizontalFlip(), # transforms.RandomVerticalFlip(), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ] )
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support