oluinioluwa814 commited on
Commit
631a933
·
verified ·
1 Parent(s): ea93eeb

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -94,11 +94,11 @@ class StableDiffusionEngine:
94
  Stable Diffusion XL Engine (HF Spaces safe version)
95
  """
96
 
97
- def __init__(
98
- self,
99
- model_id="stabilityai/stable-diffusion-xl-base-1.0",
100
- hf_token: str | None = None,
101
- ):
102
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
103
 
104
  # ✅ Proper dtype assignment
 
94
  Stable Diffusion XL Engine (HF Spaces safe version)
95
  """
96
 
97
+ def __init__(
98
+ self,
99
+ model_id="stabilityai/stable-diffusion-xl-base-1.0",
100
+ hf_token: str | None = None,
101
+ ):
102
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
103
 
104
  # ✅ Proper dtype assignment