Spaces:

dx2102
/

llama-midi

Running on Zero

dx2102 commited on 11 days ago

Commit

c0fa5a2

verified ·

1 Parent(s): b0affb5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ print('\n\n\n')
 print('Loading model...')
 pipe = transformers.pipeline(
     "text-generation",
-    model="dx2102/llama-midi-v9-2",
     torch_dtype="bfloat16",
     device_map="auto",
 )

 print('Loading model...')
 pipe = transformers.pipeline(
     "text-generation",
+    model="dx2102/llama-midi",
     torch_dtype="bfloat16",
     device_map="auto",
 )