bh4vay commited on
Commit
35e1ad5
·
verified ·
1 Parent(s): f232606

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ story_generator = load_text_model()
37
  def load_image_model():
38
  try:
39
  st.write("⏳ Loading image model...")
40
- model_id = "runwayml/stable-diffusion-v1-5"
41
  model = StableDiffusionPipeline.from_pretrained(
42
  model_id,
43
  torch_dtype=torch.float16 if device == "cuda" else torch.float32
 
37
  def load_image_model():
38
  try:
39
  st.write("⏳ Loading image model...")
40
+ model_id = "stabilityai/sd-turbo"
41
  model = StableDiffusionPipeline.from_pretrained(
42
  model_id,
43
  torch_dtype=torch.float16 if device == "cuda" else torch.float32