File size: 1,710 Bytes
948b11c
96812c9
 
948b11c
 
 
96812c9
 
 
948b11c
 
 
e7ba29d
 
 
96812c9
948b11c
 
daa5539
96812c9
 
 
948b11c
 
08476ef
e7ba29d
96812c9
 
daa5539
 
96812c9
 
6ebf2fd
 
 
 
 
 
 
 
96812c9
59a1fe9
96812c9
 
 
 
 
59a1fe9
6ebf2fd
 
 
 
59a1fe9
 
 
 
 
 
 
 
 
 
 
 
 
6ebf2fd
 
 
 
59a1fe9
 
 
 
 
 
 
6ebf2fd
59a1fe9
 
 
 
 
6ebf2fd
 
 
 
 
 
 
59a1fe9
 
96812c9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
export interface Section {
  title: string
  items: string[]
}

export interface ClassificationOutput {
  sequence: string
  labels: string[]
  scores: number[]
}

export interface WorkerMessage {
  status: 'initiate' | 'ready' | 'output' | 'complete' | 'progress' | 'error'
  progress?: number
  error?: string
  output?: any
}

export interface ZeroShotWorkerInput {
  text: string
  labels: string[]
  model: string
}

export interface TextClassificationWorkerInput {
  type: 'classify'
  text: string
  model: string
}

export type AppStatus = 'idle' | 'loading' | 'processing'

type q8 = 'q8' | 'int8' | 'bnb8' | 'uint8'
type q4 = 'q4' | 'bnb4'
type fp16 = 'fp16'
type fp32 = 'fp32'

export type QuantizationType = q8 | q4 | fp16 | fp32


export interface ModelInfo {
  id: string
  name: string
  architecture: string
  parameters: number
  likes: number
  downloads: number
  createdAt: string
  isCompatible?: boolean
  incompatibilityReason?: string
  supportedQuantizations: QuantizationType[]
  baseId?: string
}


export interface ModelInfoResponse {
  id: string
  createdAt: string
  config?: {
    architectures: string[]
    model_type: string
  }
  lastModified: string
  pipeline_tag: string
  tags: string[]
  cardData?: {
    base_model: string
  }
  baseId?: string
  transformersInfo: {
    pipeline_tag: string
    auto_model: string
    processor: string
  }
  safetensors?: {
    parameters: {
      BF16?: number
      F16?: number
      F32?: number
      total?: number
    }
  }
  siblings?: {
    rfilename: string
  }[]
  modelId?: string
  isCompatible: boolean
  incompatibilityReason?: string
  supportedQuantizations: QuantizationType[]
  likes: number
  downloads: number
}