Spaces:
Runtime error
Runtime error
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:
|