ysharma HF Staff commited on
Commit
87f9867
·
1 Parent(s): 60bca8f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,7 +31,7 @@ def monkeypatching(alpha, in_prompt): #, prompt, pipe): finetuned_lora_weights
31
  counter +=1
32
  else :
33
  tune_lora_scale(pipe.unet, alpha) #1.00)
34
- prompt = "style of sks, " + str(in_prompt) #"baby lion"
35
  image = pipe(prompt, num_inference_steps=50, guidance_scale=7).images[0]
36
  image.save("./illust_lora.jpg") #"./contents/illust_lora.jpg")
37
  return image
@@ -44,7 +44,7 @@ def accelerate_train_lora(steps):
44
  --pretrained_model_name_or_path={MODEL_NAME} \
45
  --instance_data_dir={INSTANCE_DIR} \
46
  --output_dir={OUTPUT_DIR} \
47
- --instance_prompt="style of sks" \
48
  --resolution=512 \
49
  --train_batch_size=1 \
50
  --gradient_accumulation_steps=1 \
 
31
  counter +=1
32
  else :
33
  tune_lora_scale(pipe.unet, alpha) #1.00)
34
+ prompt = "style of hclu, " + str(in_prompt) #"baby lion"
35
  image = pipe(prompt, num_inference_steps=50, guidance_scale=7).images[0]
36
  image.save("./illust_lora.jpg") #"./contents/illust_lora.jpg")
37
  return image
 
44
  --pretrained_model_name_or_path={MODEL_NAME} \
45
  --instance_data_dir={INSTANCE_DIR} \
46
  --output_dir={OUTPUT_DIR} \
47
+ --instance_prompt="style of hclu" \
48
  --resolution=512 \
49
  --train_batch_size=1 \
50
  --gradient_accumulation_steps=1 \