Hugging Face's logo

Spaces:
Collov-Labs
/
d-edit
Running on Zero

App Files Community
7
d-edit
File size: 1,104 Bytes 
d807efd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
export IMAGE_NAME="example1"
export IMAGE_NAME_2="example2"
# python main.py --name=$IMAGE_NAME \
#     --dpm="sdxl" \
#     --image \
#     --name_2=$IMAGE_NAME_2 \
#     --resolution=1024 \
#     --embedding_learning_rate=1e-4  \
#     --diffusion_model_learning_rate=5e-5 \
#     --max_emb_train_steps=500  \
#     --max_diffusion_train_steps=500 \
#     --train_batch_size=1 \
#     --gradient_accumulation_steps=5

python main.py --name=$IMAGE_NAME \
    --dpm="sdxl" \
    --image \
    --train_full_lora \
    --name_2=$IMAGE_NAME_2 \
    --resolution=1024 \
    --embedding_learning_rate=1e-4  \
    --diffusion_model_learning_rate=5e-4 \
    --max_emb_train_steps=500  \
    --max_diffusion_train_steps=500 \
    --train_batch_size=1 \
    --gradient_accumulation_steps=5


# python main.py --load_trained \
#     --dpm="sdxl" \
#     --image \
#     --name=$IMAGE_NAME \
#     --name_2=$IMAGE_NAME_2 \
#     --tgt_name=$IMAGE_NAME \
#     --guidance_scale 2.5 \
#     --edge_thickness 40 \
#     --strength 0.5 \
#     --seed 29 \
#     --num_imgs 4 \
#     --tgt_index=0 \
#     --src_index=2