amankishore commited on
Commit
cfc39b8
1 Parent(s): 81048a9

taming transformers

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +1 -2
app.py CHANGED
@@ -19,9 +19,12 @@ from run_sjc import render_one_view, tsr_stats
19
 
20
  import gradio as gr
21
  import gc
 
22
 
23
  device_glb = torch.device("cuda")
24
 
 
 
25
  def vis_routine(y, depth):
26
  pane = nerf_vis(y, depth, final_H=256)
27
  im = torch_samps_to_imgs(y)[0]
 
19
 
20
  import gradio as gr
21
  import gc
22
+ import os
23
 
24
  device_glb = torch.device("cuda")
25
 
26
+ os.system("git clone --depth 1 git@github.com:CompVis/taming-transformers.git && pip install -e taming-transformers")
27
+
28
  def vis_routine(y, depth):
29
  pane = nerf_vis(y, depth, final_H=256)
30
  im = torch_samps_to_imgs(y)[0]
requirements.txt CHANGED
@@ -13,5 +13,4 @@ transformers
13
  kornia==0.6.0
14
  git+https:///github.com/openai/CLIP.git#egg=clip
15
  imageio[ffmpeg]
16
- imageio[pyav]
17
- taming-transformers
 
13
  kornia==0.6.0
14
  git+https:///github.com/openai/CLIP.git#egg=clip
15
  imageio[ffmpeg]
16
+ imageio[pyav]