Spaces:
Running
on
Zero
Running
on
Zero
wangshuai6
commited on
Commit
·
911c3a3
1
Parent(s):
9e426da
app demo
Browse files- app.py +1 -3
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -31,7 +31,7 @@
|
|
| 31 |
# w_scheduler: src.diffusion.stateful_flow_matching.scheduling.LinearScheduler
|
| 32 |
# guidance_fn: src.diffusion.base.guidance.simple_guidance_fn
|
| 33 |
# step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
|
| 34 |
-
|
| 35 |
import torch
|
| 36 |
import argparse
|
| 37 |
from omegaconf import OmegaConf
|
|
@@ -94,8 +94,6 @@ class Pipeline:
|
|
| 94 |
images.append(image)
|
| 95 |
return images
|
| 96 |
|
| 97 |
-
import os
|
| 98 |
-
import spaces
|
| 99 |
if __name__ == "__main__":
|
| 100 |
parser = argparse.ArgumentParser()
|
| 101 |
parser.add_argument("--config", type=str, default="configs/repa_improved_ddt_xlen22de6_512.yaml")
|
|
|
|
| 31 |
# w_scheduler: src.diffusion.stateful_flow_matching.scheduling.LinearScheduler
|
| 32 |
# guidance_fn: src.diffusion.base.guidance.simple_guidance_fn
|
| 33 |
# step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
|
| 34 |
+
import os
|
| 35 |
import torch
|
| 36 |
import argparse
|
| 37 |
from omegaconf import OmegaConf
|
|
|
|
| 94 |
images.append(image)
|
| 95 |
return images
|
| 96 |
|
|
|
|
|
|
|
| 97 |
if __name__ == "__main__":
|
| 98 |
parser = argparse.ArgumentParser()
|
| 99 |
parser.add_argument("--config", type=str, default="configs/repa_improved_ddt_xlen22de6_512.yaml")
|
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
lightning==2.5.0.post0
|
| 2 |
omegaconf==2.3.0
|
| 3 |
-
torch==2.
|
| 4 |
diffusers==0.30.0
|
| 5 |
jsonargparse[signatures]>=4.27.7
|
| 6 |
accelerate
|
|
|
|
| 1 |
lightning==2.5.0.post0
|
| 2 |
omegaconf==2.3.0
|
| 3 |
+
torch==2.2.0
|
| 4 |
diffusers==0.30.0
|
| 5 |
jsonargparse[signatures]>=4.27.7
|
| 6 |
accelerate
|