Upload 3 files

README.md

@@ -1,5 +1,5 @@
 ---
-title: WaiNSFWIllustrious V110
+title: WaiNSFWIllustrious V130 Space
 emoji: 🖼
 colorFrom: purple
 colorTo: red

app.py

@@ -15,7 +15,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Make sure to use torch.float16 consistently throughout the pipeline
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "dhead/waiNSFWIllustrious_v120",
+    "dhead/waiNSFWIllustrious_v130",
     torch_dtype=torch.float16,
     variant="fp16",  # Explicitly use fp16 variant
     use_safetensors=True  # Use safetensors if available
@@ -33,7 +33,7 @@ pipe.unet.to(torch.float16)
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1216
     
-@spaces.GPU(duration=10)
+@spaces.GPU()
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     # Check and truncate prompt if too long (CLIP can only handle 77 tokens)
     if len(prompt.split()) > 60:  # Rough estimate to avoid exceeding token limit