Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
23d07c9
1
Parent(s):
25bf19b
yes
Browse files
app.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
import torch
|
2 |
import gradio as gr
|
3 |
import numpy as np
|
@@ -133,6 +134,7 @@ def encode_sdxl_prompt(prompt, negative_prompt=""):
|
|
133 |
|
134 |
# βββ Inference ββββββββββββββββββββββββββββββββββββββββββββββββ
|
135 |
@torch.no_grad()
|
|
|
136 |
def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
|
137 |
use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
|
138 |
|
|
|
1 |
+
import spaces
|
2 |
import torch
|
3 |
import gradio as gr
|
4 |
import numpy as np
|
|
|
134 |
|
135 |
# βββ Inference ββββββββββββββββββββββββββββββββββββββββββββββββ
|
136 |
@torch.no_grad()
|
137 |
+
@spaces.GPU
|
138 |
def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
|
139 |
use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
|
140 |
|