Views
No views yet

1from diffusers import DiffusionPipeline
2
3pipeline = DiffusionPipeline.from_pretrained("nroggendorff/cats")
4pipe = pipeline.to("cuda")
5
6image = pipe().images[0]
7
8image.save("cat.png")train_batch_size: 16eval_batch_size: 16num_epochs: 50gradient_accumulation_steps: 1learning_rate: 1e-4lr_warmup_steps: 500mixed_precision: "fp16"eval_metric: "mean_squared_error"