0.6b-demo

Sleeping

sumuks commited on 7 days ago

Commit

16173da

verified ·

1 Parent(s): 443499e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,8 @@ def load_model():
                 model = model_class.from_pretrained(
                     model_id,
                     torch_dtype=torch.bfloat16,
-                    device_map="auto"
                 )
                 break
             except:

                 model = model_class.from_pretrained(
                     model_id,
                     torch_dtype=torch.bfloat16,
+                    device_map="auto",
+                    subfolder="checkpoint-3717"
                 )
                 break
             except: