Need to use a public model
Browse files
    	
        app.py
    CHANGED
    
    | 
         @@ -17,7 +17,7 @@ import spaces 
     | 
|
| 17 | 
         
             
            #gradio.helpers.CACHED_FOLDER = '/data/cache'
         
     | 
| 18 | 
         | 
| 19 | 
         
             
            pipe = StableVideoDiffusionPipeline.from_pretrained(
         
     | 
| 20 | 
         
            -
                " 
     | 
| 21 | 
         
             
            )
         
     | 
| 22 | 
         
             
            pipe.to("cuda")
         
     | 
| 23 | 
         
             
            #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         
     | 
| 
         | 
|
| 17 | 
         
             
            #gradio.helpers.CACHED_FOLDER = '/data/cache'
         
     | 
| 18 | 
         | 
| 19 | 
         
             
            pipe = StableVideoDiffusionPipeline.from_pretrained(
         
     | 
| 20 | 
         
            +
                "vdo/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
         
     | 
| 21 | 
         
             
            )
         
     | 
| 22 | 
         
             
            pipe.to("cuda")
         
     | 
| 23 | 
         
             
            #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         
     |