MogensR commited on
Commit
7f13b76
·
verified ·
1 Parent(s): 878513c

Update processing/ai_background.py

Browse files
Files changed (1) hide show
  1. processing/ai_background.py +5 -1
processing/ai_background.py CHANGED
@@ -289,5 +289,9 @@ def generate_ai_background(prompt: str, width: int = 1024, height: int = 768) ->
289
 
290
  print(f"\nModule status:")
291
  print(f"- PIL Available: {PIL_AVAILABLE}")
 
292
  print(f"- Diffusers Available: {DIFFUSERS_AVAILABLE}")
293
- print(f"- Device: {generator.device}")
 
 
 
 
289
 
290
  print(f"\nModule status:")
291
  print(f"- PIL Available: {PIL_AVAILABLE}")
292
+ print(f"- PyTorch Available: {TORCH_AVAILABLE}")
293
  print(f"- Diffusers Available: {DIFFUSERS_AVAILABLE}")
294
+ if generator.pipeline:
295
+ print(f"- Device: {generator.device}")
296
+ else:
297
+ print("- AI Pipeline: Not initialized (using gradients only)")