Spaces:
Running
on
A10G
Running
on
A10G
Commit
·
2927b74
1
Parent(s):
5fc21dd
basically delirious lol
Browse files
app.py
CHANGED
@@ -126,7 +126,7 @@ pipe.to(device=DEVICE)
|
|
126 |
#pipe.unet = torch.compile(pipe.unet)
|
127 |
#pipe.vae = torch.compile(pipe.vae)
|
128 |
|
129 |
-
|
130 |
def generate_gpu(in_im_embs):
|
131 |
print('start gen')
|
132 |
in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
|
@@ -222,9 +222,7 @@ def pluck_img(user_id, user_emb):
|
|
222 |
|
223 |
|
224 |
def background_next_image():
|
225 |
-
|
226 |
-
global is_started
|
227 |
-
while True:
|
228 |
# only let it get N (maybe 3) ahead of the user
|
229 |
#not_rated_rows = prevs_df[[i[1]['user:rating'] == {' ': ' '} for i in prevs_df.iterrows()]]
|
230 |
rated_rows = prevs_df[[i[1]['user:rating'] != {' ': ' '} for i in prevs_df.iterrows()]]
|
|
|
126 |
#pipe.unet = torch.compile(pipe.unet)
|
127 |
#pipe.vae = torch.compile(pipe.vae)
|
128 |
|
129 |
+
|
130 |
def generate_gpu(in_im_embs):
|
131 |
print('start gen')
|
132 |
in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)
|
|
|
222 |
|
223 |
|
224 |
def background_next_image():
|
225 |
+
global prevs_df
|
|
|
|
|
226 |
# only let it get N (maybe 3) ahead of the user
|
227 |
#not_rated_rows = prevs_df[[i[1]['user:rating'] == {' ': ' '} for i in prevs_df.iterrows()]]
|
228 |
rated_rows = prevs_df[[i[1]['user:rating'] != {' ': ' '} for i in prevs_df.iterrows()]]
|