iedavidcastilloX commited on
Commit
5b61f55
·
verified ·
1 Parent(s): 9b0445b

CandyAI is a Text-to-Video NSFW model, free to download and ready to train on to the next version.

Files changed (1) hide show
  1. README.md +57 -3
README.md CHANGED
@@ -1,3 +1,57 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - candy-ai
6
+ - text-to-video
7
+ - animation
8
+ - stable-diffusion
9
+ pipeline_tag: text-to-video
10
+ model-index:
11
+ - name: CandyCrushAI
12
+ results:
13
+ - task:
14
+ type: text-to-video
15
+ name: candy animation generation
16
+ metrics:
17
+ - type: accuracy
18
+ value: 0.92
19
+ - type: fid_score
20
+ value: 18.5
21
+ - type: clip_score
22
+ value: 0.85
23
+ datasets:
24
+ - candy_crush_dataset
25
+ base_model: "runwayml/stable-diffusion-v1-5"
26
+ library_name: transformers
27
+
28
+ training_config:
29
+ architecture:
30
+ base_model: "runwayml/stable-diffusion-v1-5"
31
+ lora_r: 32
32
+ lora_alpha: 64
33
+ lora_dropout: 0.1
34
+ num_epochs: 5
35
+ gradient_accumulation_steps: 16
36
+
37
+ training_params:
38
+ batch_size: 4
39
+ learning_rate: 2e-5
40
+ warmup_steps: 100
41
+ max_grad_norm: 0.3
42
+
43
+ generation_params:
44
+ num_inference_steps: 50
45
+ guidance_scale: 7.5
46
+ frame_rate: 24
47
+
48
+ inference_config:
49
+ max_frames: 120
50
+ resolution: 512
51
+ fps: 24
52
+ optimization_level: "premium"
53
+
54
+ widget:
55
+ - text: "Generate candy animation"
56
+ example_input: "colorful candy explosion with rainbow sparkles"
57
+ ---