mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-26 15:51:05 +00:00
21 lines
467 B
Go
21 lines
467 B
Go
|
package tinydream
|
||
|
|
|
||
|
|
import "os"
|
||
|
|
|
||
|
|
type TinyDream struct {
|
||
|
|
assetDir string
|
||
|
|
}
|
||
|
|
|
||
|
|
func New(assetDir string) (*TinyDream, error) {
|
||
|
|
if _, err := os.Stat(assetDir); err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return &TinyDream{
|
||
|
|
assetDir: assetDir,
|
||
|
|
}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (td *TinyDream) GenerateImage(height, width, step, seed int, positive_prompt, negative_prompt, dst string) error {
|
||
|
|
return GenerateImage(height, width, step, seed, positive_prompt, negative_prompt, dst, td.assetDir)
|
||
|
|
}
|