Spaces:
Sleeping
Sleeping
Commit
•
7ac7da9
1
Parent(s):
67ec952
upweighting -- will exploit more & explore somewhat less
Browse files
app.py
CHANGED
@@ -144,7 +144,7 @@ def next_image(embs, ys, calibrate_prompts):
|
|
144 |
lin_class.coef_ = (lin_class.coef_.flatten() / (lin_class.coef_.flatten().norm())).unsqueeze(0)
|
145 |
|
146 |
rng_prompt = random.choice(prompt_list)
|
147 |
-
w = 1# if len(embs) % 2 == 0 else 0
|
148 |
im_emb = w * lin_class.coef_.to(dtype=torch.float16)
|
149 |
prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
|
150 |
print(prompt, len(ys))
|
|
|
144 |
lin_class.coef_ = (lin_class.coef_.flatten() / (lin_class.coef_.flatten().norm())).unsqueeze(0)
|
145 |
|
146 |
rng_prompt = random.choice(prompt_list)
|
147 |
+
w = 1.4# if len(embs) % 2 == 0 else 0
|
148 |
im_emb = w * lin_class.coef_.to(dtype=torch.float16)
|
149 |
prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
|
150 |
print(prompt, len(ys))
|