fffiloni commited on
Commit
62b04c4
1 Parent(s): 4efa9a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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["HF_TOKEN"]
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: