codellama-playground

Running

pcuenq HF staff commited on Aug 30, 2023

Commit

f7cdcf4

•

1 Parent(s): 3aed9a5

Revert infilling change (#10)

Files changed (1) hide show

app.py CHANGED Viewed

@@ -66,7 +66,7 @@ def generate(
             prefix, suffix = prompt.split(FIM_INDICATOR)
         except:
             raise ValueError(f"Only one {FIM_INDICATOR} allowed in prompt!")
-        # prompt = f"{FIM_PREFIX}{prefix}{FIM_SUFFIX}{suffix}{FIM_MIDDLE}"
     stream = client.generate_stream(prompt, **generate_kwargs)

             prefix, suffix = prompt.split(FIM_INDICATOR)
         except:
             raise ValueError(f"Only one {FIM_INDICATOR} allowed in prompt!")
+        prompt = f"{FIM_PREFIX}{prefix}{FIM_SUFFIX}{suffix}{FIM_MIDDLE}"
     stream = client.generate_stream(prompt, **generate_kwargs)