Lovitra commited on
Commit
d359ae9
·
verified ·
1 Parent(s): ef0b55e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ comic_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
13
  # Stable Diffusion Model
14
  model_id = "stabilityai/sd-turbo"
15
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
16
- pipe.to("cuda")
17
 
18
  # Function to generate comic strip
19
  def generate_comic(user_prompt, num_panels, art_choice):
 
13
  # Stable Diffusion Model
14
  model_id = "stabilityai/sd-turbo"
15
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, variant="fp16")
16
+ pipe.to("cpu")
17
 
18
  # Function to generate comic strip
19
  def generate_comic(user_prompt, num_panels, art_choice):