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 |

|
| 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 |

|
| 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
|
|