Spaces:
Runtime error
Runtime error
Update server.py
Browse files
server.py
CHANGED
@@ -50,7 +50,7 @@ def generate(prompt):
|
|
50 |
def sample(prompt):
|
51 |
# Sampling
|
52 |
images = (
|
53 |
-
model.sampling(prompt=prompt, top_k=256, top_p=None, softmax_temperature=1.0, num_candidates=
|
54 |
.cpu()
|
55 |
.numpy()
|
56 |
)
|
|
|
50 |
def sample(prompt):
|
51 |
# Sampling
|
52 |
images = (
|
53 |
+
model.sampling(prompt=prompt, top_k=256, top_p=None, softmax_temperature=1.0, num_candidates=9, device=device)
|
54 |
.cpu()
|
55 |
.numpy()
|
56 |
)
|