ysharma HF staff commited on
Commit
f1f55e8
·
1 Parent(s): e6d7a40
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -37,6 +37,7 @@ def monkeypatching(alpha): #, prompt, pipe): finetuned_lora_weights
37
  return image
38
 
39
  def accelerate_train_lora(steps):
 
40
  #subprocess.run(accelerate launch {"./train_lora_dreambooth.py"} \
41
  subprocess.Popen(f'accelerate launch {"./train_lora_dreambooth.py"} \
42
  --pretrained_model_name_or_path=MODEL_NAME \
@@ -50,6 +51,7 @@ def accelerate_train_lora(steps):
50
  --lr_scheduler="constant" \
51
  --lr_warmup_steps=0 \
52
  --max_train_steps={steps}' , shell=True) #30000
 
53
  return
54
 
55
  with gr.Blocks() as demo:
 
37
  return image
38
 
39
  def accelerate_train_lora(steps):
40
+ print("***********inside accelerate_train_lora 11111***********")
41
  #subprocess.run(accelerate launch {"./train_lora_dreambooth.py"} \
42
  subprocess.Popen(f'accelerate launch {"./train_lora_dreambooth.py"} \
43
  --pretrained_model_name_or_path=MODEL_NAME \
 
51
  --lr_scheduler="constant" \
52
  --lr_warmup_steps=0 \
53
  --max_train_steps={steps}' , shell=True) #30000
54
+ print("***********inside accelerate_train_lora 22222***********")
55
  return
56
 
57
  with gr.Blocks() as demo: