Spaces:
Sleeping
Sleeping
Commit
·
f867da7
1
Parent(s):
d9c0a5a
update
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ pipe.safety_checker = None
|
|
30 |
|
31 |
semaphore = Semaphore() # for preventing collisions of two simultaneous button presses
|
32 |
|
33 |
-
|
34 |
def generate_baseline(prompt, seed, steps, height_width, negative_prompt, guidance_scale, method):
|
35 |
|
36 |
semaphore.acquire()
|
|
|
30 |
|
31 |
semaphore = Semaphore() # for preventing collisions of two simultaneous button presses
|
32 |
|
33 |
+
#@spaces.GPU
|
34 |
def generate_baseline(prompt, seed, steps, height_width, negative_prompt, guidance_scale, method):
|
35 |
|
36 |
semaphore.acquire()
|
merge.py
CHANGED
@@ -1,7 +1,6 @@
|
|
1 |
import torch
|
2 |
from typing import Tuple, Callable
|
3 |
from diffusers.models.attention_processor import XFormersAttnProcessor, Attention
|
4 |
-
import xformers, xformers.ops
|
5 |
from typing import Optional
|
6 |
import math
|
7 |
import torch.nn.functional as F
|
|
|
1 |
import torch
|
2 |
from typing import Tuple, Callable
|
3 |
from diffusers.models.attention_processor import XFormersAttnProcessor, Attention
|
|
|
4 |
from typing import Optional
|
5 |
import math
|
6 |
import torch.nn.functional as F
|