github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 12:30:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix problem

Browse Source
This commit is contained in:
csunny
2023-04-29 21:50:47 +08:00
parent 0767537606
commit acf9dbbd82
3 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/model/loader.py
View File

@@ -29,8 +29,8 @@ class ModerLoader:
if debug:
print(model)
if self.device == "cuda":
model.to(self.device)
# if self.device == "cuda":
# model.to(self.device)
return model, tokenizer