Spaces:
Runtime error
Runtime error
Commit
•
7af4a09
1
Parent(s):
eecb1f6
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ make_cutouts = MakeCutouts(clip_model.visual.input_resolution, 16, 1.)
|
|
59 |
def run_all(prompt, steps, n_images, weight, clip_guided):
|
60 |
import random
|
61 |
seed = int(random.randint(0, 2147483647))
|
62 |
-
target_embed = clip_model.encode_text(clip.tokenize(prompt)).float()#.cuda()
|
63 |
|
64 |
if(clip_guided):
|
65 |
prompts = [prompt]
|
|
|
59 |
def run_all(prompt, steps, n_images, weight, clip_guided):
|
60 |
import random
|
61 |
seed = int(random.randint(0, 2147483647))
|
62 |
+
target_embed = clip_model.encode_text(clip.tokenize(prompt).to('cuda')).float()#.cuda()
|
63 |
|
64 |
if(clip_guided):
|
65 |
prompts = [prompt]
|