Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ prior_output = prior(
|
|
52 |
|
53 |
decoder.enable_model_cpu_offload()
|
54 |
decoder_output = decoder(
|
55 |
-
image_embeddings=prior_output.image_embeddings
|
56 |
prompt=prompt,
|
57 |
negative_prompt=negative_prompt,
|
58 |
guidance_scale=2.0,
|
@@ -85,7 +85,7 @@ decoder_output.save("cascade.png")
|
|
85 |
## Dataset:
|
86 |
|
87 |
**GPU used for captioning**: 1x Intel ARC A770 16GB
|
88 |
-
**Model used for captioning**: SmilingWolf
|
89 |
|
90 |
|
91 |
| dataset name | total images | total chunk |
|
|
|
52 |
|
53 |
decoder.enable_model_cpu_offload()
|
54 |
decoder_output = decoder(
|
55 |
+
image_embeddings=prior_output.image_embeddings,
|
56 |
prompt=prompt,
|
57 |
negative_prompt=negative_prompt,
|
58 |
guidance_scale=2.0,
|
|
|
85 |
## Dataset:
|
86 |
|
87 |
**GPU used for captioning**: 1x Intel ARC A770 16GB
|
88 |
+
**Model used for captioning**: SmilingWolf/wd-v1-4-convnextv2-tagger-v2
|
89 |
|
90 |
|
91 |
| dataset name | total images | total chunk |
|