jayw commited on
Commit
9d9dc1d
1 Parent(s): 6b0ae07

set unused text encoder to none

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -32,6 +32,7 @@ pipe_base.enable_model_cpu_offload()
32
  pretrained_model_path = "showlab/show-1-interpolation"
33
  pipe_interp_1 = TextToVideoIFInterpPipeline.from_pretrained(
34
  pretrained_model_path,
 
35
  torch_dtype=torch.float16,
36
  variant="fp16"
37
  )
@@ -52,6 +53,7 @@ pipe_sr_1_image.enable_model_cpu_offload()
52
  pretrained_model_path = "showlab/show-1-sr1"
53
  pipe_sr_1_cond = TextToVideoIFSuperResolutionPipeline_Cond.from_pretrained(
54
  pretrained_model_path,
 
55
  torch_dtype=torch.float16
56
  )
57
  pipe_sr_1_cond.enable_model_cpu_offload()
@@ -313,5 +315,4 @@ with gr.Blocks(css=css) as demo:
313
 
314
  share_button.click(None, [], [], _js=share_js)
315
 
316
- demo.queue(max_size=12).launch(show_api=True, share=True)
317
-
 
32
  pretrained_model_path = "showlab/show-1-interpolation"
33
  pipe_interp_1 = TextToVideoIFInterpPipeline.from_pretrained(
34
  pretrained_model_path,
35
+ text_encoder=None,
36
  torch_dtype=torch.float16,
37
  variant="fp16"
38
  )
 
53
  pretrained_model_path = "showlab/show-1-sr1"
54
  pipe_sr_1_cond = TextToVideoIFSuperResolutionPipeline_Cond.from_pretrained(
55
  pretrained_model_path,
56
+ text_encoder=None,
57
  torch_dtype=torch.float16
58
  )
59
  pipe_sr_1_cond.enable_model_cpu_offload()
 
315
 
316
  share_button.click(None, [], [], _js=share_js)
317
 
318
+ demo.queue(max_size=12).launch(show_api=True)