Remove some unusued bits. (#1067)

This commit is contained in:
Laurent Mazare
2023-10-09 20:49:57 +02:00
committed by GitHub
parent 4d04ac83c7
commit b34d7f0248
5 changed files with 2 additions and 18 deletions

View File

@ -1,5 +1,3 @@
#![allow(unused)]
#[cfg(feature = "accelerate")]
extern crate accelerate_src;
@ -10,7 +8,7 @@ use candle_transformers::models::stable_diffusion;
use candle_transformers::models::wuerstchen;
use anyhow::{Error as E, Result};
use candle::{DType, Device, IndexOp, Module, Tensor, D};
use candle::{DType, Device, IndexOp, Tensor};
use clap::Parser;
use tokenizers::Tokenizer;
@ -80,14 +78,6 @@ struct Args {
/// The file specifying the tokenizer to used for prior tokenization.
prior_tokenizer: Option<String>,
/// The size of the sliced attention or 0 for automatic slicing (disabled by default)
#[arg(long)]
sliced_attention_size: Option<usize>,
/// The number of steps to run the diffusion for.
#[arg(long, default_value_t = 30)]
n_steps: usize,
/// The number of samples to generate.
#[arg(long, default_value_t = 1)]
num_samples: i64,
@ -220,10 +210,8 @@ fn run(args: Args) -> Result<()> {
cpu,
height,
width,
n_steps,
tokenizer,
final_image,
sliced_attention_size,
num_samples,
clip_weights,
prior_weights,