black image

app.py

@@ -115,7 +115,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
             seed = random.randint(0, MAX_SEED)
         
         progress(0.3, desc="Setting up generator...")
-        generator = torch.Generator().manual_seed(seed)
+        generator = torch.Generator("cuda").manual_seed(seed)  # Explicitly use CUDA generator
         
         progress(0.4, desc="Generating image...")
         with torch.autocast('cuda'):
@@ -125,14 +125,15 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
                 height=height,
                 num_inference_steps=num_inference_steps, 
                 generator=generator,
-                guidance_scale=5.0,
-                max_sequence_length=512
+                guidance_scale=7.5,  # Increased guidance scale
+                max_sequence_length=2048
             ).images[0]
 
         torch.cuda.empty_cache()  # Clean up GPU memory after generation
         progress(1.0, desc="Done!")
         return image, seed
     except Exception as e:
+        print(f"Error in infer: {str(e)}")  # Add detailed error logging
         return None, f"Error generating image: {str(e)}"
  
 examples = [
@@ -192,7 +193,7 @@ def create_interface():
 
         with gr.Column(elem_id="col-container"):
             gr.Markdown(f"""# Text to Product
-            Using Mistral + Flux + Trellis
+            Using Mistral-7B-Instruct-v0.3 + FLUX.1-dev + Trellis
             """)
             
             with gr.Row():