Skip to content

Instantly share code, notes, and snippets.

@zacc
zacc / gist:40f618b76bd26db2664409b76ed74cff
Last active August 9, 2021 19:32
run transformers gpt-2 locally to test output
from simpletransformers.language_generation import LanguageGenerationModel, LanguageGenerationArgs
def main():
model_path = "models/<model_folder>/"
model = LanguageGenerationModel("gpt2", model_path, args={'fp16': False, 'cache_dir': f"{model_path}.cache/"}, use_cuda=False)
args = {
'max_length': 1000,
@zacc
zacc / kill_all.py
Last active January 3, 2016 10:38 — forked from bebef1987/kill_all.py
def kill_all(self):
self.marionette.switch_to_frame()
self.marionette.execute_async_script("""
// Kills all running apps, except the homescreen.
function killAll() {
let originsToClose = [];
let that = this;
let runningApps = window.wrappedJSObject.AppWindowManager.getRunningApps();
for (let origin in runningApps) {