Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
6f70ac0
1
Parent(s):
fddb899
- app.py +2 -1
- requirements.txt +1 -0
app.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import spaces
|
|
|
2 |
import torch
|
3 |
import gradio as gr
|
4 |
import numpy as np
|
@@ -135,8 +136,8 @@ def encode_sdxl_prompt(prompt, negative_prompt=""):
|
|
135 |
}
|
136 |
|
137 |
# βββ Inference ββββββββββββββββββββββββββββββββββββββββββββββββ
|
138 |
-
@torch.no_grad()
|
139 |
@spaces.GPU
|
|
|
140 |
def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
|
141 |
use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
|
142 |
|
|
|
1 |
import spaces
|
2 |
+
|
3 |
import torch
|
4 |
import gradio as gr
|
5 |
import numpy as np
|
|
|
136 |
}
|
137 |
|
138 |
# βββ Inference ββββββββββββββββββββββββββββββββββββββββββββββββ
|
|
|
139 |
@spaces.GPU
|
140 |
+
@torch.no_grad()
|
141 |
def infer(prompt, negative_prompt, adapter_l_file, adapter_g_file, strength, noise, gate_prob,
|
142 |
use_anchor, steps, cfg_scale, scheduler_name, width, height, seed):
|
143 |
|
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
sentencepiece
|
2 |
accelerate
|
3 |
diffusers
|
|
|
1 |
+
spaces
|
2 |
sentencepiece
|
3 |
accelerate
|
4 |
diffusers
|