Skip to content

GitLab

Commit 92e15272 authored by chenpangpang's avatar chenpangpang
Browse files

feat: update

parent 8e0fac1d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
CineDiffusion/app.py
View file @ 92e15272
import gradio as gr import gradio as gr
import numpy as np import numpy as np
import spaces
import torch import torch
from diffusers import DiffusionPipeline from diffusers import DiffusionPipeline
from numpy.random import PCG64DXSM, Generator, SeedSequence from numpy.random import PCG64DXSM, Generator, SeedSequence
...@@ -64,7 +63,6 @@ def get_random_seed() -> int: ...@@ -64,7 +63,6 @@ def get_random_seed() -> int:
return int(rng.integers(0, MAX_SEED)) return int(rng.integers(0, MAX_SEED))
@spaces.GPU()
def infer( def infer(
prompt: str, prompt: str,
aspect_ratio: str, aspect_ratio: str,
......
CineDiffusion/requirements.txt
View file @ 92e15272
xformers
sentencepiece sentencepiece
peft peft
invisible_watermark
protobuf
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment