Commit
·
3fa2808
1
Parent(s):
6cbc9fd
End of training
Browse files- README.md +5 -5
- unet/diffusion_pytorch_model.safetensors +1 -1
- val_imgs_grid.png +0 -0
README.md
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
license: creativeml-openrail-m
|
| 4 |
base_model: CompVis/stable-diffusion-v1-4
|
| 5 |
datasets:
|
| 6 |
-
- MaxReynolds/
|
| 7 |
tags:
|
| 8 |
- stable-diffusion
|
| 9 |
- stable-diffusion-diffusers
|
|
@@ -14,7 +14,7 @@ inference: true
|
|
| 14 |
|
| 15 |
# Text-to-image finetuning - MaxReynolds/MyStreamlitModel
|
| 16 |
|
| 17 |
-
This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **MaxReynolds/
|
| 18 |
|
| 19 |

|
| 20 |
|
|
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
|
|
| 28 |
import torch
|
| 29 |
|
| 30 |
pipeline = DiffusionPipeline.from_pretrained("MaxReynolds/MyStreamlitModel", torch_dtype=torch.float16)
|
| 31 |
-
prompt = "
|
| 32 |
image = pipeline(prompt).images[0]
|
| 33 |
image.save("my_image.png")
|
| 34 |
```
|
|
@@ -37,7 +37,7 @@ image.save("my_image.png")
|
|
| 37 |
|
| 38 |
These are the key hyperparameters used during training:
|
| 39 |
|
| 40 |
-
* Epochs:
|
| 41 |
* Learning rate: 1e-05
|
| 42 |
* Batch size: 1
|
| 43 |
* Gradient accumulation steps: 4
|
|
@@ -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/
|
|
|
|
| 3 |
license: creativeml-openrail-m
|
| 4 |
base_model: CompVis/stable-diffusion-v1-4
|
| 5 |
datasets:
|
| 6 |
+
- MaxReynolds/MyPatternDataset
|
| 7 |
tags:
|
| 8 |
- stable-diffusion
|
| 9 |
- stable-diffusion-diffusers
|
|
|
|
| 14 |
|
| 15 |
# Text-to-image finetuning - MaxReynolds/MyStreamlitModel
|
| 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/MyStreamlitModel", torch_dtype=torch.float16)
|
| 31 |
+
prompt = "<r4nd0m-l4b3l>"
|
| 32 |
image = pipeline(prompt).images[0]
|
| 33 |
image.save("my_image.png")
|
| 34 |
```
|
|
|
|
| 37 |
|
| 38 |
These are the key hyperparameters used during training:
|
| 39 |
|
| 40 |
+
* Epochs: 42
|
| 41 |
* Learning rate: 1e-05
|
| 42 |
* Batch size: 1
|
| 43 |
* Gradient accumulation steps: 4
|
|
|
|
| 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/su62a0vp).
|
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:ee361f0b0007d7fafd324340dffd8d6222384791ab107aa23a3995e900224f67
|
| 3 |
size 3438167536
|
val_imgs_grid.png
CHANGED
|
|