MaxReynolds
commited on
Commit
•
25403b8
1
Parent(s):
e6eeb43
End of training
Browse files- README.md +3 -3
- unet/diffusion_pytorch_model.safetensors +1 -1
- val_imgs_grid.png +0 -0
README.md
CHANGED
@@ -14,7 +14,7 @@ inference: true
|
|
14 |
|
15 |
# Text-to-image finetuning - MaxReynolds/MyPatternModel
|
16 |
|
17 |
-
This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **MaxReynolds/MyPatternDataset** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['
|
18 |
|
19 |
![val_imgs_grid](./val_imgs_grid.png)
|
20 |
|
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
|
|
28 |
import torch
|
29 |
|
30 |
pipeline = DiffusionPipeline.from_pretrained("MaxReynolds/MyPatternModel", torch_dtype=torch.float16)
|
31 |
-
prompt = "
|
32 |
image = pipeline(prompt).images[0]
|
33 |
image.save("my_image.png")
|
34 |
```
|
@@ -45,4 +45,4 @@ These are the key hyperparameters used during training:
|
|
45 |
* Mixed-precision: fp16
|
46 |
|
47 |
|
48 |
-
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/max-f-reynolds/text2image-fine-tune/runs/
|
|
|
14 |
|
15 |
# Text-to-image finetuning - MaxReynolds/MyPatternModel
|
16 |
|
17 |
+
This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **MaxReynolds/MyPatternDataset** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: [',<r4nd0m-l4b3l>']:
|
18 |
|
19 |
![val_imgs_grid](./val_imgs_grid.png)
|
20 |
|
|
|
28 |
import torch
|
29 |
|
30 |
pipeline = DiffusionPipeline.from_pretrained("MaxReynolds/MyPatternModel", torch_dtype=torch.float16)
|
31 |
+
prompt = ",<r4nd0m-l4b3l>"
|
32 |
image = pipeline(prompt).images[0]
|
33 |
image.save("my_image.png")
|
34 |
```
|
|
|
45 |
* Mixed-precision: fp16
|
46 |
|
47 |
|
48 |
+
More information on all the CLI arguments and the environment are available on your [`wandb` run page](https://wandb.ai/max-f-reynolds/text2image-fine-tune/runs/8v2yfdkf).
|
unet/diffusion_pytorch_model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3438167536
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:371c140ab76858085f1a81995f7e78a11e19deee8d93481373a8a6b7b98f4364
|
3 |
size 3438167536
|
val_imgs_grid.png
CHANGED