Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import os
|
|
| 3 |
import subprocess
|
| 4 |
from huggingface_hub import snapshot_download
|
| 5 |
|
| 6 |
-
hf_token = os.environ.get
|
| 7 |
print(hf_token)
|
| 8 |
|
| 9 |
def set_accelerate_default_config():
|
|
@@ -40,7 +40,8 @@ def train_dreambooth_lora_sdxl(instance_data_dir):
|
|
| 40 |
"--max_train_steps=25",
|
| 41 |
"--checkpointing_steps=717",
|
| 42 |
"--seed=0",
|
| 43 |
-
"--push_to_hub"
|
|
|
|
| 44 |
]
|
| 45 |
|
| 46 |
try:
|
|
|
|
| 3 |
import subprocess
|
| 4 |
from huggingface_hub import snapshot_download
|
| 5 |
|
| 6 |
+
hf_token = os.environ.get("HF_TOKEN")
|
| 7 |
print(hf_token)
|
| 8 |
|
| 9 |
def set_accelerate_default_config():
|
|
|
|
| 40 |
"--max_train_steps=25",
|
| 41 |
"--checkpointing_steps=717",
|
| 42 |
"--seed=0",
|
| 43 |
+
"--push_to_hub",
|
| 44 |
+
f"--hub_token={hf_token}"
|
| 45 |
]
|
| 46 |
|
| 47 |
try:
|