mlpc-lab
/

TokenCompose_SD14_B

StableDiffusionPipeline

stable-diffusion

image-generation

compositionality

Model card Files Files and versions

zwcolin commited on Dec 4, 2023

Commit

03f5427

·

1 Parent(s): 9af6c08

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -19,7 +19,7 @@ We strongly recommend using the [🤗Diffuser](https://github.com/huggingface/di
 import torch
 from diffusers import StableDiffusionPipeline
-model_id = "mlpc-lab/TokenCompose_SD14_A"
 device = "cuda"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)

 import torch
 from diffusers import StableDiffusionPipeline
+model_id = "mlpc-lab/TokenCompose_SD14_B"
 device = "cuda"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)