Skip to content

Commit

Permalink
Fix model
Browse files Browse the repository at this point in the history
  • Loading branch information
collindutter committed May 6, 2024
1 parent 9e4e999 commit d363deb
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions docs/griptape-framework/engines/image-generation-engines.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ from griptape.rules import Ruleset, Rule
# Create a driver configured to use Stable Diffusion via Bedrock.
driver = AmazonBedrockImageGenerationDriver(
image_generation_model_driver=BedrockStableDiffusionImageGenerationModelDriver(),
model="stability.stable-diffusion-xl-v0",
model="stability.stable-diffusion-xl-v1",
)

# Create an engine configured to use the driver.
Expand Down Expand Up @@ -52,7 +52,7 @@ from griptape.drivers import AmazonBedrockImageGenerationDriver, \
# Create a driver configured to use Stable Diffusion via Bedrock.
driver = AmazonBedrockImageGenerationDriver(
image_generation_model_driver=BedrockStableDiffusionImageGenerationModelDriver(),
model="stability.stable-diffusion-xl-v0",
model="stability.stable-diffusion-xl-v1",
)

# Create an engine configured to use the driver.
Expand All @@ -78,7 +78,7 @@ from griptape.loaders import ImageLoader
# Create a driver configured to use Stable Diffusion via Bedrock.
driver = AmazonBedrockImageGenerationDriver(
image_generation_model_driver=BedrockStableDiffusionImageGenerationModelDriver(),
model="stability.stable-diffusion-xl-v0",
model="stability.stable-diffusion-xl-v1",
)

# Create an engine configured to use the driver.
Expand Down Expand Up @@ -109,7 +109,7 @@ from griptape.loaders import ImageLoader
# Create a driver configured to use Stable Diffusion via Bedrock.
driver = AmazonBedrockImageGenerationDriver(
image_generation_model_driver=BedrockStableDiffusionImageGenerationModelDriver(),
model="stability.stable-diffusion-xl-v0",
model="stability.stable-diffusion-xl-v1",
)

# Create an engine configured to use the driver.
Expand Down Expand Up @@ -143,7 +143,7 @@ from griptape.loaders import ImageLoader
# Create a driver configured to use Stable Diffusion via Bedrock.
driver = AmazonBedrockImageGenerationDriver(
image_generation_model_driver=BedrockStableDiffusionImageGenerationModelDriver(),
model="stability.stable-diffusion-xl-v0",
model="stability.stable-diffusion-xl-v1",
)

# Create an engine configured to use the driver.
Expand Down

0 comments on commit d363deb

Please sign in to comment.