feat: update

92e15272 · chenpangpang · 8e0fac1d · 92e15272 · 92e15272
Commit 92e15272 authored Dec 04, 2024 by chenpangpang
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

CineDiffusion/app.py CineDiffusion/app.py +0 -2

CineDiffusion/requirements.txt CineDiffusion/requirements.txt +3 -0

No files found.
--- a/CineDiffusion/app.py
+++ b/CineDiffusion/app.py
 import gradio as gr
 import numpy as np
-import spaces
 import torch
 from diffusers import DiffusionPipeline
 from numpy.random import PCG64DXSM, Generator, SeedSequence
@@ -64,7 +63,6 @@ def get_random_seed() -> int:
    return int(rng.integers(0, MAX_SEED))


-@spaces.GPU()
 def infer(
        prompt: str,
        aspect_ratio: str,

--- a/CineDiffusion/requirements.txt
+++ b/CineDiffusion/requirements.txt
+xformers
 sentencepiece
 peft
+invisible_watermark
+protobuf
\ No newline at end of file