bigscience-workshop / bigscience-workshop/petals

How to parallelize this code for model.generate?

Open
#427 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
10.6k
Forks
650
PR merge metrics
No merged PRs in 30d

Description

As the title says, how can I parallelize this?

```
def generate_output(row):
inputs = tokenizer(prompt, return_tensors="pt")["input_ids"]
outputs = model.generate(inputs, max_new_tokens=185, temperature=0.0, eos_token_id=tokenizer.encode("}")[0])
result = tokenizer.decode(outputs[0])
completion = extract_completion(result)

for index, row in df.iterrows():
generate_output(row)
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.