Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import warnings
|
|
| 6 |
warnings.filterwarnings('ignore')
|
| 7 |
|
| 8 |
# Constants
|
| 9 |
-
MODEL_ID = "
|
| 10 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 11 |
DTYPE = torch.float32 # Using float32 for better compatibility
|
| 12 |
|
|
@@ -21,7 +21,7 @@ class FooocusGenerator:
|
|
| 21 |
self.pipe = StableDiffusionPipeline.from_pretrained(
|
| 22 |
MODEL_ID,
|
| 23 |
torch_dtype=DTYPE,
|
| 24 |
-
use_auth_token=
|
| 25 |
)
|
| 26 |
|
| 27 |
if DEVICE == "cuda":
|
|
|
|
| 6 |
warnings.filterwarnings('ignore')
|
| 7 |
|
| 8 |
# Constants
|
| 9 |
+
MODEL_ID = "CompVis/stable-diffusion-v1-4" # Using v1.4 which is publicly available
|
| 10 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 11 |
DTYPE = torch.float32 # Using float32 for better compatibility
|
| 12 |
|
|
|
|
| 21 |
self.pipe = StableDiffusionPipeline.from_pretrained(
|
| 22 |
MODEL_ID,
|
| 23 |
torch_dtype=DTYPE,
|
| 24 |
+
use_auth_token=False # No auth token needed for v1.4
|
| 25 |
)
|
| 26 |
|
| 27 |
if DEVICE == "cuda":
|