How to use dragostom24/StarCoder2_FT with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("bigcode/starcoder2-3b", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("dragostom24/StarCoder2_FT") prompt = "-" image = pipe(prompt).images[0]
https://huggingface.co/dragostom24/StarCoder2_FT/resolve/main/README.md
hf download hf://dragostom24/StarCoder2_FT/README.md
curl -L -o README.md https://huggingface.co/dragostom24/StarCoder2_FT/resolve/main/README.md
tags: - text-to-image - lora - diffusers - template:diffusion-lora widget: - output: url: images/starcoder2_metrics_plot.png text: '-' base_model: bigcode/starcoder2-3b instance_prompt: null license: unknown
Download them in the Files & versions tab.
