Prgckwb commited on
Commit
3a215cd
1 Parent(s): 414fa11

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -19
README.md CHANGED
@@ -66,28 +66,21 @@ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
66
 
67
  ## Trigger words
68
 
69
- You should use an icon of trpfrog to trigger the image generation.
70
 
71
- ## Download model
72
-
73
- Weights for this model are available in Safetensors format.
74
-
75
- [Download](Prgckwb/trpfrog-sdxl-lora-3/tree/main) them in the Files & versions tab.
76
-
77
-
78
-
79
- ## Intended uses & limitations
80
-
81
- #### How to use
82
 
83
  ```python
84
- # TODO: add an example code snippet for running this diffusion pipeline
85
- ```
86
 
87
- #### Limitations and bias
 
88
 
89
- [TODO: provide examples of latent issues and potential remediations]
 
 
90
 
91
- ## Training details
92
-
93
- [TODO: describe the data used to train the model]
 
66
 
67
  ## Trigger words
68
 
69
+ You should use an icon of "trpfrog" to trigger the image generation.
70
 
71
+ ## How to use
 
 
 
 
 
 
 
 
 
 
72
 
73
  ```python
74
+ from diffusers import DiffusionPipeline
75
+ import torch
76
 
77
+ base_model_id = 'stabilityai/stable-diffusion-xl-base-1.0'
78
+ lora_model_id = 'Prgckwb/trpfrog-diffusion-sdxl-lora'
79
 
80
+ pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16)
81
+ pipe = pipe.to("cuda")
82
+ pipe.load_lora_weights(lora_model_id)
83
 
84
+ image = pipe("trpfrog", num_inference_steps=25).images[0]
85
+ image.save('trpfrog.png')
86
+ ```