Habana
/

stable-diffusion

Model card Files Files and versions

regisss HF Staff commited on Apr 24, 2023

Commit

7c08f5d

·

1 Parent(s): 343f91b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ Learn more about how to take advantage of the power of Habana HPUs to train and
 ## Stable Diffusion HPU configuration
-This model only contains the `GaudiConfig` file for running **Stable Diffusion 1** (e.g. [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4)) or **Stable Diffusion 2** (e.g. [stabilityai/stable-diffusion-2](https://huggingface.co/stabilityai/stable-diffusion-2)) on Habana's Gaudi processors (HPU).
 **This model contains no model weights, only a GaudiConfig.**
@@ -30,7 +30,7 @@ from optimum.habana import GaudiConfig
 from optimum.habana.diffusers import GaudiDDIMScheduler, GaudiStableDiffusionPipeline
-model_name = "stabilityai/stable-diffusion-2"
 scheduler = GaudiDDIMScheduler.from_pretrained(model_name, subfolder="scheduler")

 ## Stable Diffusion HPU configuration
+This model only contains the `GaudiConfig` file for running **Stable Diffusion v1** (e.g. [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5)) on Habana's Gaudi processors (HPU).
 **This model contains no model weights, only a GaudiConfig.**
 from optimum.habana.diffusers import GaudiDDIMScheduler, GaudiStableDiffusionPipeline
+model_name = "runwayml/stable-diffusion-v1-5"
 scheduler = GaudiDDIMScheduler.from_pretrained(model_name, subfolder="scheduler")