Spaces:

cmu-gil
/

LegoGPT-Demo

Running on A100

AvaLovelace commited on about 15 hours ago

Commit

eabc75d

1 Parent(s): 6a3d39e

Update default max_regenerations

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,8 +39,8 @@ def main():
         print('Running in Hugging Face Space, setting up environment...')
         setup()
-    model_cfg = LegoGPTConfig()
-    model = LegoGPT(LegoGPTConfig())
     def generate_lego(
             prompt: str,

         print('Running in Hugging Face Space, setting up environment...')
         setup()
+    model_cfg = LegoGPTConfig(max_regenerations=10)
+    model = LegoGPT(model_cfg)
     def generate_lego(
             prompt: str,