Update to latent-gcode diffusion model with SD 1.5

app.py

@@ -24,10 +24,12 @@ def get_generator():
         dtype = torch.float16 if device == "cuda" else torch.float32
         
         print("Loading Stable Diffusion pipeline...")
+        # Use SD 1.5 which is more reliably available
         pipe = StableDiffusionPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-2-1-base",
+            "runwayml/stable-diffusion-v1-5",
             torch_dtype=dtype,
             safety_checker=None,
+            use_safetensors=True,
         ).to(device)
         
         print("Loading gcode decoder...")

requirements.txt

@@ -4,3 +4,4 @@ transformers>=4.36
 diffusers>=0.25
 accelerate>=0.25
 huggingface_hub>=0.20
+safetensors>=0.4