Spaces:
Runtime error
Runtime error
File size: 258 Bytes
ec840f2 |
1 2 3 4 5 |
import torch
from transformers import pipeline
generate_text = pipeline(model="databricks/dolly-v2-12b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
generate_text("Explain to me the difference between nuclear fission and fusion.") |