BAAI
/

Bunny-Llama-3-8B-V

Text Generation

Model card Files Files and versions Community

BoyaWu10 commited on Apr 29

Commit

8f4c803

•

1 Parent(s): a65ec98

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -47,6 +47,7 @@ warnings.filterwarnings('ignore')
 # set device
 device = 'cuda'  # or cpu
 # create model
 model = AutoModelForCausalLM.from_pretrained(

 # set device
 device = 'cuda'  # or cpu
+torch.set_default_device(device)
 # create model
 model = AutoModelForCausalLM.from_pretrained(