Spaces:
Running
on
Zero
Running
on
Zero
wangshuai6
commited on
Commit
·
ed7275e
1
Parent(s):
e3ecee1
init
Browse files
app.py
CHANGED
|
@@ -33,6 +33,7 @@
|
|
| 33 |
# step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
|
| 34 |
import random
|
| 35 |
import os
|
|
|
|
| 36 |
import torch
|
| 37 |
import argparse
|
| 38 |
from omegaconf import OmegaConf
|
|
@@ -43,7 +44,7 @@ from src.diffusion.flow_matching.scheduling import LinearScheduler
|
|
| 43 |
from PIL import Image
|
| 44 |
import gradio as gr
|
| 45 |
import tempfile
|
| 46 |
-
|
| 47 |
|
| 48 |
from huggingface_hub import snapshot_download
|
| 49 |
|
|
|
|
| 33 |
# step_fn: src.diffusion.stateful_flow_matching.sampling.ode_step_fn
|
| 34 |
import random
|
| 35 |
import os
|
| 36 |
+
import spaces
|
| 37 |
import torch
|
| 38 |
import argparse
|
| 39 |
from omegaconf import OmegaConf
|
|
|
|
| 44 |
from PIL import Image
|
| 45 |
import gradio as gr
|
| 46 |
import tempfile
|
| 47 |
+
|
| 48 |
|
| 49 |
from huggingface_hub import snapshot_download
|
| 50 |
|