From d3f05eae8c4f2df186b46e433be101ac39fceca5 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Sun, 10 Sep 2023 09:40:27 +0100
Subject: Move some models to candle-transformers so that it's easier to
 re-use. (#794)

* Move some models to candle-transformers so that they can be shared.

* Also move falcon.

* Move Llama.

* Move whisper (partial).
---
 candle-examples/examples/bigcode/main.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'candle-examples/examples/bigcode/main.rs')

diff --git a/candle-examples/examples/bigcode/main.rs b/candle-examples/examples/bigcode/main.rs
index 652cd47f..3540f75d 100644
--- a/candle-examples/examples/bigcode/main.rs
+++ b/candle-examples/examples/bigcode/main.rs
@@ -7,8 +7,7 @@ extern crate accelerate_src;
 use anyhow::{Error as E, Result};
 use clap::Parser;
 
-mod model;
-use model::{Config, GPTBigCode};
+use candle_transformers::models::bigcode::{Config, GPTBigCode};
 
 use candle::{DType, Device, Tensor};
 use candle_nn::VarBuilder;
-- 
cgit v1.2.3