Instructions to use yuting89830/GAN_AI_hw5 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use yuting89830/GAN_AI_hw5 with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("yuting89830/GAN_AI_hw5", torch_dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| { | |
| "image_dir": "public_data/images", | |
| "output_dir": "steps50000_bs16_lr1e-4_seed42_no_inf100_inferbs32_ema", | |
| "image_size": 256, | |
| "batch_size": 16, | |
| "num_epochs": 100, | |
| "max_steps": 50000, | |
| "lr": 0.0001, | |
| "save_every": 5000, | |
| "sample_every": 1000, | |
| "num_preview_samples": 4, | |
| "preview_inference_steps": 30, | |
| "resume_from_checkpoint": null, | |
| "seed": 42, | |
| "num_workers": 4, | |
| "mixed_precision": "no", | |
| "ema_decay": 0.9999, | |
| "no_ema": false | |
| } |