Added safe tensors and memory efficient attention

app.py

@@ -7,6 +7,7 @@ model_id = "dream-textures/texture-diffusion"
 if torch.cuda.is_available():
     pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
     pipe = pipe.to("cuda")
+    pipe.enable_xformers_memory_efficient_attention()
 else:
     pipe = StableDiffusionPipeline.from_pretrained(model_id)
 
@@ -18,6 +19,6 @@ with streamlit.form(key="Interpolation"):
     numSteps = streamlit.slider(label="Number of Inference steps", value=50, min_value=1, max_value=1000)
     numImages = streamlit.slider(label="Number of Generated Images", value=1, min_value=1, max_value=10)
     guidanceScale = streamlit.slider(label="Guidance Scale", value=7.5, min_value=0.0, max_value=100.0)
-    image = pipe(prompt,  num_inference_steps=numSteps, num_images_per_prompt=numImages, guidance_scale=guidanceScale).images[0]
+    images = pipe(prompt=prompt, num_inference_steps=numSteps, num_images_per_prompt=numImages, guidance_scale=guidanceScale).images
     streamlit.form_submit_button("Interpolate")
-    synthesizedImage = streamlit.image(image)
+    synthesizedImage = streamlit.image(images)

requirements.txt

@@ -1,4 +1,6 @@
 torch
+accelerate
 transformers
 diffusers
-streamlit
+streamlit
+safetensors