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
|