BestWishYsh commited on
Commit
288a0e7
·
verified ·
1 Parent(s): c72f944

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import CLIPTextModel, CLIPTokenizer
14
  from utils.unet import UNet3DConditionModel
15
  from utils.pipeline_magictime import MagicTimePipeline
16
  from utils.util import save_videos_grid, convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint, load_diffusers_lora_unet, convert_ldm_clip_text_model
17
- # import spaces
18
 
19
  pretrained_model_path = "./ckpts/Base_Model/stable-diffusion-v1-5"
20
  inference_config_path = "./sample_configs/RealisticVision.yaml"
@@ -62,7 +62,7 @@ examples = [
62
  print(f"### Cleaning cached examples ...")
63
  os.system(f"rm -rf gradio_cached_examples/")
64
 
65
- # @spaces.GPU(duration=300)
66
  class MagicTimeController:
67
  def __init__(self):
68
 
 
14
  from utils.unet import UNet3DConditionModel
15
  from utils.pipeline_magictime import MagicTimePipeline
16
  from utils.util import save_videos_grid, convert_ldm_unet_checkpoint, convert_ldm_clip_checkpoint, convert_ldm_vae_checkpoint, load_diffusers_lora_unet, convert_ldm_clip_text_model
17
+ import spaces
18
 
19
  pretrained_model_path = "./ckpts/Base_Model/stable-diffusion-v1-5"
20
  inference_config_path = "./sample_configs/RealisticVision.yaml"
 
62
  print(f"### Cleaning cached examples ...")
63
  os.system(f"rm -rf gradio_cached_examples/")
64
 
65
+ @spaces.GPU(duration=300)
66
  class MagicTimeController:
67
  def __init__(self):
68