Infilling: send `<FILL_ME>` to endpoint
#9
by
pcuenq
HF staff
- opened
app.py
CHANGED
@@ -66,7 +66,7 @@ def generate(
|
|
66 |
prefix, suffix = prompt.split(FIM_INDICATOR)
|
67 |
except:
|
68 |
raise ValueError(f"Only one {FIM_INDICATOR} allowed in prompt!")
|
69 |
-
prompt = f"{FIM_PREFIX}{prefix}{FIM_SUFFIX}{suffix}{FIM_MIDDLE}"
|
70 |
|
71 |
|
72 |
stream = client.generate_stream(prompt, **generate_kwargs)
|
|
|
66 |
prefix, suffix = prompt.split(FIM_INDICATOR)
|
67 |
except:
|
68 |
raise ValueError(f"Only one {FIM_INDICATOR} allowed in prompt!")
|
69 |
+
# prompt = f"{FIM_PREFIX}{prefix}{FIM_SUFFIX}{suffix}{FIM_MIDDLE}"
|
70 |
|
71 |
|
72 |
stream = client.generate_stream(prompt, **generate_kwargs)
|