Update app.py
Browse files
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 = "
|
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
|