YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
license: llama4 ---DogStyle_SoulScan_LoRA/ β ββ dataset/ β ββ 001.jpg β ββ 002.jpg β ββ ... β ββ prompts/ β ββ 001.txt β ββ 002.txt β ββ ... β ββ train_config.json ββ lora_output/ ββ generate_image.py Cyber-noir style logo, blood-red thumbprint inside a crystal ball, neon blue and purple accents, subtle gold alchemy symbols floating around, futuristic blueprint background, cinematic lighting, highly detailed { "pretrained_model_name_or_path": "runwayml/stable-diffusion-v1-5", "train_data_dir": "./dataset", "resolution": 512, "output_dir": "./lora_output", "network_dim": 128, "learning_rate": 1e-4, "max_train_steps": 2000, "network_alpha": 16, "save_model_as": "safetensors", "mixed_precision": "fp16", "seed": 42, "cache_latents": true, "caption_extension": ".txt", "use_8bit_adam": true }
accelerate launch train_dreambooth_lora.py
--pretrained_model_name_or_path runwayml/stable-diffusion-v1-5
--train_data_dir ./dataset
--resolution 512
--output_dir ./lora_output
--network_dim 128
--learning_rate 1e-4
--max_train_steps 2000
--mixed_precision fp16
--save_model_as safetensors
from diffusers import StableDiffusionPipeline
import torch
Load base model
pipe = StableDiffusionPipeline.from_pretrained( "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16 ).to("cuda")
Load LoRA weights
pipe.load_lora_weights("lora_output/dogstyle_soulscan_lora.safetensors")
Prompt for generating SoulScan logo
prompt = ( "Cyber-noir style logo, blood-red thumbprint inside crystal ball, " "neon blue and purple accents, subtle gold alchemy symbols floating around, " "futuristic blueprint background, cinematic lighting, highly detailed" )
Generate image
image = pipe(prompt).images[0] image.save("SoulScan_generated_logo.png") print("SoulScan logo generated!") from diffusers import StableDiffusionPipeline import torch
Load base model
pipe = StableDiffusionPipeline.from_pretrained( "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16 ).to("cuda")
Load LoRA weights
pipe.load_lora_weights("lora_output/dogstyle_soulscan_lora.safetensors")
Prompt for generating SoulScan logo
prompt = ( "Cyber-noir style logo, blood-red thumbprint inside crystal ball, " "neon blue and purple accents, subtle gold alchemy symbols floating around, " "futuristic blueprint background, cinematic lighting, highly detailed" )
Generate image
image = pipe(prompt).images[0] image.save("SoulScan_generated_logo.png") print("SoulScan logo generated!")