Spaces:
Running
on
Zero
Running
on
Zero
add zerogpu support
Browse files
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import os
|
| 3 |
import sys
|
|
@@ -30,7 +31,7 @@ def download_model():
|
|
| 30 |
if not os.path.exists(local_file):
|
| 31 |
hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_1024_v1/', force_download=True)
|
| 32 |
|
| 33 |
-
|
| 34 |
def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
|
| 35 |
resolution = (576, 1024)
|
| 36 |
download_model()
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
import os
|
| 4 |
import sys
|
|
|
|
| 31 |
if not os.path.exists(local_file):
|
| 32 |
hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_1024_v1/', force_download=True)
|
| 33 |
|
| 34 |
+
@spaces.GPU
|
| 35 |
def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123):
|
| 36 |
resolution = (576, 1024)
|
| 37 |
download_model()
|