nikigoli's picture
Removed GPU dependency from main model code
ef94a96 verified