Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
wip
Browse files
src/lib/components/InferencePlayground/InferencePlayground.svelte
CHANGED
|
@@ -32,17 +32,12 @@
|
|
| 32 |
}
|
| 33 |
];
|
| 34 |
|
| 35 |
-
$: if (conversations.length > 1) {
|
| 36 |
-
viewCode = false;
|
| 37 |
-
}
|
| 38 |
-
|
| 39 |
let systemMessage: ChatCompletionInputMessage = { role: 'system', content: '' };
|
| 40 |
let hfToken: string | null = import.meta.env.VITE_HF_TOKEN;
|
| 41 |
let viewCode = false;
|
| 42 |
let showTokenModal = false;
|
| 43 |
let showModelPickerModal = false;
|
| 44 |
let loading = false;
|
| 45 |
-
let tokens = 0;
|
| 46 |
let latency = 0;
|
| 47 |
let abortControllers: AbortController[] = [];
|
| 48 |
let waitForNonStreaming = true;
|
|
@@ -96,11 +91,6 @@
|
|
| 96 |
});
|
| 97 |
}
|
| 98 |
|
| 99 |
-
function deleteConversation(idx: number) {
|
| 100 |
-
deleteAndGetItem(conversations, idx);
|
| 101 |
-
conversations = conversations;
|
| 102 |
-
}
|
| 103 |
-
|
| 104 |
function reset() {
|
| 105 |
systemMessage.content = '';
|
| 106 |
conversations = conversations.map((conversation) => {
|
|
@@ -123,10 +113,6 @@
|
|
| 123 |
async function runInference(conversation: Conversation) {
|
| 124 |
const startTime = performance.now();
|
| 125 |
const hf = createHfInference(hfToken);
|
| 126 |
-
const requestMessages = [
|
| 127 |
-
...(systemPromptSupported && systemMessage?.content?.length ? [systemMessage] : []),
|
| 128 |
-
...conversation.messages
|
| 129 |
-
];
|
| 130 |
|
| 131 |
if (conversation.streaming) {
|
| 132 |
const streamingMessage = { role: 'assistant', content: '' };
|
|
@@ -262,22 +248,18 @@
|
|
| 262 |
? '*:w-1/3'
|
| 263 |
: '*:w-full'} dark:divide-gray-800"
|
| 264 |
>
|
| 265 |
-
|
| 266 |
-
|
| 267 |
-
|
| 268 |
-
|
| 269 |
-
|
| 270 |
-
|
| 271 |
-
|
| 272 |
-
|
| 273 |
-
|
| 274 |
-
|
| 275 |
-
|
| 276 |
-
|
| 277 |
-
on:deleteMessage={(e) => deleteMessage(e.detail)}
|
| 278 |
-
on:deleteConversation={(e) => deleteConversation(e.detail)}
|
| 279 |
-
/>
|
| 280 |
-
{/each}
|
| 281 |
</div>
|
| 282 |
<div
|
| 283 |
class="fixed inset-x-0 bottom-0 flex h-20 items-center gap-2 overflow-hidden whitespace-nowrap px-3 md:absolute"
|
|
|
|
| 32 |
}
|
| 33 |
];
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
let systemMessage: ChatCompletionInputMessage = { role: 'system', content: '' };
|
| 36 |
let hfToken: string | null = import.meta.env.VITE_HF_TOKEN;
|
| 37 |
let viewCode = false;
|
| 38 |
let showTokenModal = false;
|
| 39 |
let showModelPickerModal = false;
|
| 40 |
let loading = false;
|
|
|
|
| 41 |
let latency = 0;
|
| 42 |
let abortControllers: AbortController[] = [];
|
| 43 |
let waitForNonStreaming = true;
|
|
|
|
| 91 |
});
|
| 92 |
}
|
| 93 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 94 |
function reset() {
|
| 95 |
systemMessage.content = '';
|
| 96 |
conversations = conversations.map((conversation) => {
|
|
|
|
| 113 |
async function runInference(conversation: Conversation) {
|
| 114 |
const startTime = performance.now();
|
| 115 |
const hf = createHfInference(hfToken);
|
|
|
|
|
|
|
|
|
|
|
|
|
| 116 |
|
| 117 |
if (conversation.streaming) {
|
| 118 |
const streamingMessage = { role: 'assistant', content: '' };
|
|
|
|
| 248 |
? '*:w-1/3'
|
| 249 |
: '*:w-full'} dark:divide-gray-800"
|
| 250 |
>
|
| 251 |
+
<Conversation
|
| 252 |
+
{loading}
|
| 253 |
+
conversation={conversations[0]}
|
| 254 |
+
index={0}
|
| 255 |
+
{viewCode}
|
| 256 |
+
on:addMessage={addMessage}
|
| 257 |
+
on:messageValueChanged={(e) => {
|
| 258 |
+
const { conversationIdx, messageIdx, value } = e.detail;
|
| 259 |
+
updateMessage(value, conversationIdx, messageIdx);
|
| 260 |
+
}}
|
| 261 |
+
on:deleteMessage={(e) => deleteMessage(e.detail)}
|
| 262 |
+
/>
|
|
|
|
|
|
|
|
|
|
|
|
|
| 263 |
</div>
|
| 264 |
<div
|
| 265 |
class="fixed inset-x-0 bottom-0 flex h-20 items-center gap-2 overflow-hidden whitespace-nowrap px-3 md:absolute"
|
src/lib/components/InferencePlayground/InferencePlaygroundConversation.svelte
CHANGED
|
@@ -2,20 +2,16 @@
|
|
| 2 |
import { createEventDispatcher } from 'svelte';
|
| 3 |
import CodeSnippets from './InferencePlaygroundCodeSnippets.svelte';
|
| 4 |
import Message from './InferencePlaygroundMessage.svelte';
|
| 5 |
-
import PlaygroundOptions from './InferencePlaygroundGenerationConfig.svelte';
|
| 6 |
import IconPlus from '../Icons/IconPlus.svelte';
|
| 7 |
import type { Conversation } from '$lib/types';
|
| 8 |
|
| 9 |
export let loading;
|
| 10 |
export let conversation: Conversation;
|
| 11 |
-
export let index;
|
| 12 |
export let viewCode;
|
| 13 |
-
export let sideBySide = false;
|
| 14 |
|
| 15 |
const dispatch = createEventDispatcher<{
|
| 16 |
addMessage: void;
|
| 17 |
deleteMessage: number;
|
| 18 |
-
deleteConversation: number;
|
| 19 |
}>();
|
| 20 |
|
| 21 |
let messageContainer: HTMLDivElement | null = null;
|
|
@@ -39,49 +35,15 @@
|
|
| 39 |
class:animate-pulse={loading && !conversation.streaming}
|
| 40 |
bind:this={messageContainer}
|
| 41 |
>
|
| 42 |
-
{#if sideBySide}
|
| 43 |
-
<div
|
| 44 |
-
class="sticky top-0 flex h-11 flex-none items-center gap-2 whitespace-nowrap rounded-lg border border-gray-200/80 bg-white pl-3 pr-2 text-sm leading-none shadow-sm *:flex-none dark:border-gray-800 dark:bg-gray-800/70 dark:hover:bg-gray-800"
|
| 45 |
-
class:mr-3={index === 0}
|
| 46 |
-
class:mx-3={index === 1}
|
| 47 |
-
>
|
| 48 |
-
<div class="size-3.5 rounded bg-black dark:bg-gray-400"></div>
|
| 49 |
-
<div>{conversation.model}</div>
|
| 50 |
-
<button
|
| 51 |
-
class="ml-auto flex size-6 items-center justify-center rounded bg-gray-50 text-xs hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700"
|
| 52 |
-
on:click={() => dispatch('deleteConversation', index)}
|
| 53 |
-
>
|
| 54 |
-
✕
|
| 55 |
-
</button>
|
| 56 |
-
<button
|
| 57 |
-
class="group relative flex size-6 items-center justify-center rounded bg-gray-50 hover:bg-gray-100 dark:bg-gray-800 dark:hover:bg-gray-700"
|
| 58 |
-
>
|
| 59 |
-
<svg xmlns="http://www.w3.org/2000/svg" width="1em" height="1em" viewBox="0 0 32 32"
|
| 60 |
-
><path
|
| 61 |
-
fill="currentColor"
|
| 62 |
-
d="M27 16.76v-1.53l1.92-1.68A2 2 0 0 0 29.3 11l-2.36-4a2 2 0 0 0-1.73-1a2 2 0 0 0-.64.1l-2.43.82a11.35 11.35 0 0 0-1.31-.75l-.51-2.52a2 2 0 0 0-2-1.61h-4.68a2 2 0 0 0-2 1.61l-.51 2.52a11.48 11.48 0 0 0-1.32.75l-2.38-.86A2 2 0 0 0 6.79 6a2 2 0 0 0-1.73 1L2.7 11a2 2 0 0 0 .41 2.51L5 15.24v1.53l-1.89 1.68A2 2 0 0 0 2.7 21l2.36 4a2 2 0 0 0 1.73 1a2 2 0 0 0 .64-.1l2.43-.82a11.35 11.35 0 0 0 1.31.75l.51 2.52a2 2 0 0 0 2 1.61h4.72a2 2 0 0 0 2-1.61l.51-2.52a11.48 11.48 0 0 0 1.32-.75l2.42.82a2 2 0 0 0 .64.1a2 2 0 0 0 1.73-1l2.28-4a2 2 0 0 0-.41-2.51ZM25.21 24l-3.43-1.16a8.86 8.86 0 0 1-2.71 1.57L18.36 28h-4.72l-.71-3.55a9.36 9.36 0 0 1-2.7-1.57L6.79 24l-2.36-4l2.72-2.4a8.9 8.9 0 0 1 0-3.13L4.43 12l2.36-4l3.43 1.16a8.86 8.86 0 0 1 2.71-1.57L13.64 4h4.72l.71 3.55a9.36 9.36 0 0 1 2.7 1.57L25.21 8l2.36 4l-2.72 2.4a8.9 8.9 0 0 1 0 3.13L27.57 20Z"
|
| 63 |
-
/><path
|
| 64 |
-
fill="currentColor"
|
| 65 |
-
d="M16 22a6 6 0 1 1 6-6a5.94 5.94 0 0 1-6 6Zm0-10a3.91 3.91 0 0 0-4 4a3.91 3.91 0 0 0 4 4a3.91 3.91 0 0 0 4-4a3.91 3.91 0 0 0-4-4Z"
|
| 66 |
-
/></svg
|
| 67 |
-
>
|
| 68 |
-
<PlaygroundOptions
|
| 69 |
-
bind:conversation
|
| 70 |
-
classNames="absolute top-8 right-0 w-56 invisible group-focus:visible hover:visible border border-gray-200/80 bg-white z-10 px-4 py-6 text-sm shadow-sm dark:border-gray-800 dark:bg-gray-800 rounded-xl"
|
| 71 |
-
/>
|
| 72 |
-
</button>
|
| 73 |
-
</div>
|
| 74 |
-
{/if}
|
| 75 |
{#if !viewCode}
|
| 76 |
{#each conversation.messages as message, messageIdx}
|
| 77 |
<Message
|
| 78 |
class="border-b"
|
| 79 |
{message}
|
| 80 |
-
conversationIdx={index}
|
| 81 |
{messageIdx}
|
| 82 |
on:messageValueChanged
|
| 83 |
on:delete={() => dispatch('deleteMessage', messageIdx)}
|
| 84 |
-
autofocus={!
|
| 85 |
/>
|
| 86 |
{/each}
|
| 87 |
|
|
|
|
| 2 |
import { createEventDispatcher } from 'svelte';
|
| 3 |
import CodeSnippets from './InferencePlaygroundCodeSnippets.svelte';
|
| 4 |
import Message from './InferencePlaygroundMessage.svelte';
|
|
|
|
| 5 |
import IconPlus from '../Icons/IconPlus.svelte';
|
| 6 |
import type { Conversation } from '$lib/types';
|
| 7 |
|
| 8 |
export let loading;
|
| 9 |
export let conversation: Conversation;
|
|
|
|
| 10 |
export let viewCode;
|
|
|
|
| 11 |
|
| 12 |
const dispatch = createEventDispatcher<{
|
| 13 |
addMessage: void;
|
| 14 |
deleteMessage: number;
|
|
|
|
| 15 |
}>();
|
| 16 |
|
| 17 |
let messageContainer: HTMLDivElement | null = null;
|
|
|
|
| 35 |
class:animate-pulse={loading && !conversation.streaming}
|
| 36 |
bind:this={messageContainer}
|
| 37 |
>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
{#if !viewCode}
|
| 39 |
{#each conversation.messages as message, messageIdx}
|
| 40 |
<Message
|
| 41 |
class="border-b"
|
| 42 |
{message}
|
|
|
|
| 43 |
{messageIdx}
|
| 44 |
on:messageValueChanged
|
| 45 |
on:delete={() => dispatch('deleteMessage', messageIdx)}
|
| 46 |
+
autofocus={!loading && messageIdx === conversation.messages.length - 1}
|
| 47 |
/>
|
| 48 |
{/each}
|
| 49 |
|
src/lib/components/InferencePlayground/InferencePlaygroundMessage.svelte
CHANGED
|
@@ -3,13 +3,12 @@
|
|
| 3 |
import { type ChatCompletionInputMessage } from '@huggingface/tasks';
|
| 4 |
|
| 5 |
export let message: ChatCompletionInputMessage;
|
| 6 |
-
export let conversationIdx: number;
|
| 7 |
export let messageIdx: number;
|
| 8 |
export let autofocus: boolean = false;
|
| 9 |
|
| 10 |
const dispatch = createEventDispatcher<{
|
| 11 |
delete: void;
|
| 12 |
-
messageValueChanged: {
|
| 13 |
}>();
|
| 14 |
</script>
|
| 15 |
|
|
@@ -22,8 +21,7 @@
|
|
| 22 |
<textarea
|
| 23 |
{autofocus}
|
| 24 |
value={message.content}
|
| 25 |
-
on:input={(e) =>
|
| 26 |
-
dispatch('messageValueChanged', { conversationIdx, messageIdx, value: e.target.value })}
|
| 27 |
placeholder="Enter {message.role} message"
|
| 28 |
class="resize-none rounded bg-transparent px-2 py-2.5 ring-gray-100 [field-sizing:content] hover:resize-y hover:bg-white focus:resize-y focus:bg-white focus:ring group-hover/message:ring @2xl:px-3 dark:ring-gray-600 dark:hover:bg-gray-900 dark:focus:bg-gray-900"
|
| 29 |
rows="1"
|
|
|
|
| 3 |
import { type ChatCompletionInputMessage } from '@huggingface/tasks';
|
| 4 |
|
| 5 |
export let message: ChatCompletionInputMessage;
|
|
|
|
| 6 |
export let messageIdx: number;
|
| 7 |
export let autofocus: boolean = false;
|
| 8 |
|
| 9 |
const dispatch = createEventDispatcher<{
|
| 10 |
delete: void;
|
| 11 |
+
messageValueChanged: { messageIdx: number; value: string };
|
| 12 |
}>();
|
| 13 |
</script>
|
| 14 |
|
|
|
|
| 21 |
<textarea
|
| 22 |
{autofocus}
|
| 23 |
value={message.content}
|
| 24 |
+
on:input={(e) => dispatch('messageValueChanged', { messageIdx, value: e.target.value })}
|
|
|
|
| 25 |
placeholder="Enter {message.role} message"
|
| 26 |
class="resize-none rounded bg-transparent px-2 py-2.5 ring-gray-100 [field-sizing:content] hover:resize-y hover:bg-white focus:resize-y focus:bg-white focus:ring group-hover/message:ring @2xl:px-3 dark:ring-gray-600 dark:hover:bg-gray-900 dark:focus:bg-gray-900"
|
| 27 |
rows="1"
|