No tienes acceso a esta clase

¡Continúa aprendiendo! Únete y comienza a potenciar tu carrera

Probando demo de generación de imágenes

13/16
Recursos

Aportes 4

Preguntas 3

Ordenar por:

¿Quieres ver más aportes, preguntas y respuestas de la comunidad?

Estuve recibiendo el error relacionado al 'use_auth_token', estuve investigando al parecer el error se debe al cambio que se esta haciendo en el tipo de autenticación de Hugging Face, La única solución que encontré fue pasar la propiedad como False y además descargar desde el Hub las demás propiedades del modelo y enviarlas directamente, Así quedo mi archivo utils.py:

import json

import numpy as np
import torch
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
from huggingface_hub import hf_hub_download

CONFIG_NAME = "config.json"
revision = None
cache_dir = None
force_download = False
proxies = None
resume_download = False
local_files_only = False
token = None


def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512"):
    """
    Loads a pre-trained LightweightGAN model from Hugging Face Model Hub.

    Args:
        model_name (str): The name of the pre-trained model to load. Defaults to "ceyda/butterfly_cropped_uniq1K_512".
        model_version (str): The version of the pre-trained model to load. Defaults to None.

    Returns:
        LightweightGAN: The loaded pre-trained model.
    """
    # Load the config
    config_file = hf_hub_download(
        repo_id=str(model_name),
        filename=CONFIG_NAME,
        revision=revision,
        cache_dir=cache_dir,
        force_download=force_download,
        proxies=proxies,
        resume_download=resume_download,
        token=token,
        local_files_only=local_files_only,
    )
    with open(config_file, "r", encoding="utf-8") as f:
        config = json.load(f)

    # Call the _from_pretrained with all the needed arguments
    gan = LightweightGAN(latent_dim=256, image_size=512)

    gan = gan._from_pretrained(
        model_id=str(model_name),
        revision=revision,
        cache_dir=cache_dir,
        force_download=force_download,
        proxies=proxies,
        resume_download=resume_download,
        local_files_only=local_files_only,
        token=token,
        use_auth_token=False,
        config=config,  # usually in **model_kwargs
    )

    gan.eval()
    return gan


def genera(gan, batch_size=1):
    """
    Generates images using the given GAN model.

    Args:
        gan (nn.Module): The GAN model to use for generating images.
        batch_size (int, optional): The number of images to generate in each batch. Defaults to 1.

    Returns:
        numpy.ndarray: A numpy array of generated images.
    """
    with torch.no_grad():
        ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
        ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
    return ims

Puden encontar mi space corriendo aqui: https://huggingface.co/spaces/RaymundoSGlz/mariposas

![](https://static.platzi.com/media/user_upload/image-2fdf5a1c-6d09-4b2a-9935-b8b33e0da640.jpg) tal cual que mis compañeros , me solicita un token

siguiendo los pasos del curso utilizando visual code para la generacion de codigo y Anaconda.Navigator indica el error que muestra en la url agradezco su colaboracion

Para quienes tienen el error de autenticación, lo que deben hacer es lo siguiente: 1\. Generar un Acces Token con el rol Write. Esto lo hacen en el su perfil de HF\