Spaces:

thenativefox
/

RAG

Sleeping

App Files Files Community

RAG / sentence-transformers_all-MiniLM-L6-v2 /fixed_chunks /_accelerate.txt_chunk_5.txt

thenativefox

Added split files and tables

939262b 11 months ago

history blame contribute delete

232 Bytes

	device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")

	model.to(device)

	train_dataloader, eval_dataloader, model, optimizer = accelerator.prepare(

	train_dataloader, eval_dataloader, model, optimizer
	)