Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -100,7 +100,7 @@ def get_resolution(resolution):
|
|
| 100 |
def load_models(model_path, enable_compile=False, attention_backend="native"):
|
| 101 |
print(f"Loading models from {model_path}...")
|
| 102 |
|
| 103 |
-
use_auth_token =
|
| 104 |
|
| 105 |
if not os.path.exists(model_path):
|
| 106 |
vae = AutoencoderKL.from_pretrained(
|
|
|
|
| 100 |
def load_models(model_path, enable_compile=False, attention_backend="native"):
|
| 101 |
print(f"Loading models from {model_path}...")
|
| 102 |
|
| 103 |
+
use_auth_token = True if HF_TOKEN else False
|
| 104 |
|
| 105 |
if not os.path.exists(model_path):
|
| 106 |
vae = AutoencoderKL.from_pretrained(
|