Fabrice-TIERCELIN commited on
Commit
6638ae9
1 Parent(s): bd6a6a5

Add the imports of SUPIR

Browse files
Files changed (1) hide show
  1. GenVideo_app.py +16 -2
GenVideo_app.py CHANGED
@@ -1,12 +1,26 @@
 
 
1
  import gradio as gr
 
 
 
 
2
  import torch
 
 
 
 
 
 
 
 
 
 
3
  from diffusers import AnimateDiffPipeline, DDIMScheduler, MotionAdapter
4
  from diffusers.utils import export_to_gif
5
  from diffusers.utils import export_to_video
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  import uuid
8
- import spaces
9
- from huggingface_hub import hf_hub_download
10
 
11
  hf_hub_download(repo_id="laion/CLIP-ViT-bigG-14-laion2B-39B-b160k", filename="open_clip_pytorch_model.bin", local_dir="laion_CLIP-ViT-bigG-14-laion2B-39B-b160k")
12
  hf_hub_download(repo_id="camenduru/SUPIR", filename="sd_xl_base_1.0_0.9vae.safetensors", local_dir="yushan777_SUPIR")
 
1
+ import os
2
+
3
  import gradio as gr
4
+ from gradio_imageslider import ImageSlider
5
+ import argparse
6
+ from SUPIR.util import HWC3, upscale_image, fix_resize, convert_dtype
7
+ import numpy as np
8
  import torch
9
+ from SUPIR.util import create_SUPIR_model, load_QF_ckpt
10
+ from PIL import Image
11
+ from llava.llava_agent import LLavaAgent
12
+ from CKPT_PTH import LLAVA_MODEL_PATH
13
+ import einops
14
+ import copy
15
+ import time
16
+ import spaces
17
+ from huggingface_hub import hf_hub_download
18
+
19
  from diffusers import AnimateDiffPipeline, DDIMScheduler, MotionAdapter
20
  from diffusers.utils import export_to_gif
21
  from diffusers.utils import export_to_video
22
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
23
  import uuid
 
 
24
 
25
  hf_hub_download(repo_id="laion/CLIP-ViT-bigG-14-laion2B-39B-b160k", filename="open_clip_pytorch_model.bin", local_dir="laion_CLIP-ViT-bigG-14-laion2B-39B-b160k")
26
  hf_hub_download(repo_id="camenduru/SUPIR", filename="sd_xl_base_1.0_0.9vae.safetensors", local_dir="yushan777_SUPIR")