Spaces:

dx2102
/

llama-midi

Running on Zero

dx2102 commited on Mar 31

Commit

36bdd2d

verified ·

1 Parent(s): a53a488

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ print('Loading model...')
 pipe = transformers.pipeline(
     "text-generation",
     model="dx2102/llama-midi",
-    revision="c303c108399aba837146e893375849b918f413b3",
     torch_dtype="bfloat16",
     device="cuda",
 )

 pipe = transformers.pipeline(
     "text-generation",
     model="dx2102/llama-midi",
+    # revision="c303c108399aba837146e893375849b918f413b3",
     torch_dtype="bfloat16",
     device="cuda",
 )