rdeinla commited on
Commit
e340179
·
verified ·
1 Parent(s): b2e812a

Model card auto-generated by SimpleTuner

Browse files
Files changed (1) hide show
  1. README.md +9 -9
README.md CHANGED
@@ -31,10 +31,10 @@ A pirate ship heading out to sea
31
  ## Validation settings
32
  - CFG: `7.5`
33
  - CFG Rescale: `0.0`
34
- - Steps: `20`
35
  - Sampler: `None`
36
  - Seed: `42`
37
- - Resolution: `512`
38
 
39
  Note: The validation settings are not necessarily the same as the [training settings](#training-settings).
40
 
@@ -53,8 +53,8 @@ You may reuse the base model text encoder for inference.
53
  - Training steps: 5
54
  - Learning rate: 0.00105
55
  - Max grad norm: 0.01
56
- - Effective batch size: 8
57
- - Micro-batch size: 8
58
  - Gradient accumulation steps: 1
59
  - Number of GPUs: 1
60
  - Prediction type: flow-matching
@@ -73,8 +73,8 @@ You may reuse the base model text encoder for inference.
73
 
74
  ### wikiart_s
75
  - Repeats: 1
76
- - Total number of images: 36
77
- - Total number of aspect buckets: 3
78
  - Resolution: 1.0 megapixels
79
  - Cropped: False
80
  - Crop style: None
@@ -100,10 +100,10 @@ pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps
100
  image = pipeline(
101
  prompt=prompt,
102
  negative_prompt=negative_prompt,
103
- num_inference_steps=20,
104
  generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
105
- width=512,
106
- height=512,
107
  guidance_scale=7.5,
108
  ).images[0]
109
  image.save("output.png", format="PNG")
 
31
  ## Validation settings
32
  - CFG: `7.5`
33
  - CFG Rescale: `0.0`
34
+ - Steps: `10`
35
  - Sampler: `None`
36
  - Seed: `42`
37
+ - Resolution: `1024`
38
 
39
  Note: The validation settings are not necessarily the same as the [training settings](#training-settings).
40
 
 
53
  - Training steps: 5
54
  - Learning rate: 0.00105
55
  - Max grad norm: 0.01
56
+ - Effective batch size: 1
57
+ - Micro-batch size: 1
58
  - Gradient accumulation steps: 1
59
  - Number of GPUs: 1
60
  - Prediction type: flow-matching
 
73
 
74
  ### wikiart_s
75
  - Repeats: 1
76
+ - Total number of images: 40
77
+ - Total number of aspect buckets: 7
78
  - Resolution: 1.0 megapixels
79
  - Cropped: False
80
  - Crop style: None
 
100
  image = pipeline(
101
  prompt=prompt,
102
  negative_prompt=negative_prompt,
103
+ num_inference_steps=10,
104
  generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
105
+ width=1024,
106
+ height=1024,
107
  guidance_scale=7.5,
108
  ).images[0]
109
  image.save("output.png", format="PNG")