Update README.md
Browse files
README.md
CHANGED
@@ -85,7 +85,7 @@ prompt = "Anthropomorphic cat dressed as a pilot"
|
|
85 |
negative_prompt = ""
|
86 |
|
87 |
with torch.cuda.amp.autocast(dtype=dtype):
|
88 |
-
prior_output =
|
89 |
prompt=prompt,
|
90 |
height=1024,
|
91 |
width=1024,
|
@@ -93,7 +93,7 @@ with torch.cuda.amp.autocast(dtype=dtype):
|
|
93 |
guidance_scale=4.0,
|
94 |
num_images_per_prompt=num_images_per_prompt,
|
95 |
)
|
96 |
-
decoder_output =
|
97 |
image_embeddings=prior_output.image_embeddings,
|
98 |
prompt=prompt,
|
99 |
negative_prompt=negative_prompt,
|
|
|
85 |
negative_prompt = ""
|
86 |
|
87 |
with torch.cuda.amp.autocast(dtype=dtype):
|
88 |
+
prior_output = prior(
|
89 |
prompt=prompt,
|
90 |
height=1024,
|
91 |
width=1024,
|
|
|
93 |
guidance_scale=4.0,
|
94 |
num_images_per_prompt=num_images_per_prompt,
|
95 |
)
|
96 |
+
decoder_output = decoder(
|
97 |
image_embeddings=prior_output.image_embeddings,
|
98 |
prompt=prompt,
|
99 |
negative_prompt=negative_prompt,
|